philipturner / metal-flash-attentionLinks
FlashAttention (Metal Port)
☆572Updated last year
Alternatives and similar repositories for metal-flash-attention
Users that are interested in metal-flash-attention are comparing it to the libraries listed below
Sorting:
- Efficient framework-agnostic data loading☆455Updated 3 months ago
- ☆576Updated last year
- ☆219Updated 11 months ago
- Benchmark of Apple MLX operations on all Apple Silicon chips (GPU, CPU) + MPS and CUDA.☆209Updated last week
- Large Language Models (LLMs) applications and tools running on Apple Silicon in real-time with Apple MLX.☆459Updated 11 months ago
- 1.58 Bit LLM on Apple Silicon using MLX☆237Updated last year
- Python bindings for ggml☆146Updated last year
- CLI to demonstrate running a large language model (LLM) on Apple Neural Engine.☆117Updated last year
- Run transformers (incl. LLMs) on the Apple Neural Engine.☆63Updated 2 years ago
- Fast parallel LLM inference for MLX☆241Updated last year
- C API for MLX☆158Updated this week
- Official implementation of Half-Quadratic Quantization (HQQ)☆905Updated 3 weeks ago
- On-device Image Generation for Apple Silicon☆683Updated 9 months ago
- LLM training in simple, raw C/Metal Shading Language☆60Updated last year
- Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".☆279Updated 2 years ago
- LLM-based code completion engine☆190Updated 11 months ago
- Code for paper: "QuIP: 2-Bit Quantization of Large Language Models With Guarantees"☆393Updated last year
- ModernBERT model optimized for Apple Neural Engine.☆29Updated last year
- Inference of Mamba models in pure C☆196Updated last year
- Convert StableHLO models into Apple Core ML format☆21Updated 3 weeks ago
- Phi-3.5 for Mac: Locally-run Vision and Language Models for Apple Silicon☆274Updated 2 months ago
- ☆58Updated 2 years ago
- VPTQ, A Flexible and Extreme low-bit quantization algorithm☆671Updated 8 months ago
- scalable and robust tree-based speculative decoding algorithm☆366Updated 11 months ago
- Fast low-bit matmul kernels in Triton☆418Updated 3 weeks ago
- Apple GPU microarchitecture☆571Updated last year
- FastMLX is a high performance production ready API to host MLX models.☆339Updated 9 months ago
- An efficent implementation of the method proposed in "The Era of 1-bit LLMs"☆155Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆592Updated 5 months ago
- Profile your CoreML models directly from Python 🐍☆29Updated 4 months ago