https://github.com/manishklach/mlx-metal-kernels

Experimental MLX custom Metal kernels for Apple Silicon — fast attention, decode, KV-cache, and future Mac GPU inference primitives.

apple-gpu apple-silicon attention custom-kernels deep-learning flashattention gpu-kernels inference kv-cache llm llm-inference machine-learning macos metal metal-kernels mlx mps python transformers

Last synced: 01 Jul 2026