AI & ML interests

search every byte

erikkaumย 
posted an update 6 months ago
view post
Post
3,268
Releasing my first kernel ๐Ÿ”ฅ MaxSim

Late-interaction retrieval (ColBERT / PyLate) bottlenecks on materializing the full similarity matrix. This kernel avoids it by using tiled scoring with simdgroup_matrix (Metal) and WMMA.

The result is 3โ€“5ร— speedup compared to naive PyTorch baseline ๐Ÿ”ฅ

Benchmarks:
- SmallRerank (B=32, C=10): up to 3.2ร— (M3 Pro) / 2.8ร— (A100)
- HeavyRerank (B=32, C=100): up to 3.8ร— (M3 Pro) / 5.3ร— (A100)
- LongDocStress (Ld=1024): up to 6.2ร— (L4)

Try it out ๐Ÿ‘‡
https://huggingface.co/kernels/erikkaum/maxsim
erikkaumย 
posted an update over 1 year ago
view post
Post
2,717
ZML just released a technical preview of their new Inference Engine: LLMD.

- Just 2.4GB container, which means fast startup times and efficient autoscaling
- Cross-Platform GPU Support: works on both NVIDIA and AMD GPUs.
- written in Zig

I just tried it out and deployed it on Hugging Face Inference Endpoints and wrote a quick guide ๐Ÿ‘‡ You can try it in like 5 minutes!

https://huggingface.co/blog/erikkaum/test-driving-llmd-inference-engine
  • 1 reply
ยท
erikkaumย 
posted an update over 1 year ago
view post
Post
2,216
We just released native support for @SGLang and @vllm-project in Inference Endpoints ๐Ÿ”ฅ

Inference Endpoints is becoming the central place where you deploy high performance Inference Engines.

And that provides the managed infra for it. Instead of spending weeks configuring infrastructure, managing servers, and debugging deployment issues, you can focus on what matters most: your AI model and your users ๐Ÿ™Œ
erikkaumย 
posted an update almost 2 years ago
view post
Post
1,818
A while ago I started experimenting with compiling the Python interpreter to WASM.

To build a secure, fast, and lightweight sandbox for code execution โ€” ideal for running LLM-generated Python code.

- Send code simply as a POST request
- 1-2ms startup times

Hack away:
https://github.com/ErikKaum/runner