apple / ml-recurrent-drafterLinks
☆214Updated 5 months ago
Alternatives and similar repositories for ml-recurrent-drafter
Users that are interested in ml-recurrent-drafter are comparing it to the libraries listed below
Sorting:
- ☆173Updated this week
- scalable and robust tree-based speculative decoding algorithm☆349Updated 5 months ago
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)☆156Updated this week
- Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".☆277Updated last year
- A high-throughput and memory-efficient inference and serving engine for LLMs☆264Updated 9 months ago
- A safetensors extension to efficiently store sparse quantized tensors on disk☆135Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆255Updated this week
- KV cache compression for high-throughput LLM inference☆132Updated 5 months ago
- Load compute kernels from the Hub☆203Updated this week
- Fast low-bit matmul kernels in Triton☆330Updated this week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆205Updated this week
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆128Updated 7 months ago
- ring-attention experiments☆144Updated 8 months ago
- [ICLR'25] Fast Inference of MoE Models with CPU-GPU Orchestration☆218Updated 7 months ago
- Efficient LLM Inference over Long Sequences☆382Updated 2 weeks ago
- [ICML 2024] CLLMs: Consistency Large Language Models☆396Updated 7 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆80Updated 10 months ago
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆359Updated 2 weeks ago
- Code for "LayerSkip: Enabling Early Exit Inference and Self-Speculative Decoding", ACL 2024☆318Updated 2 months ago
- Code for data-aware compression of DeepSeek models☆36Updated last month
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆121Updated 7 months ago
- [NeurIPS 2024] KVQuant: Towards 10 Million Context Length LLM Inference with KV Cache Quantization☆360Updated 11 months ago
- Applied AI experiments and examples for PyTorch☆281Updated last month
- Benchmark suite for LLMs from Fireworks.ai☆76Updated last week
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆64Updated 3 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆224Updated 7 months ago
- A Quirky Assortment of CuTe Kernels☆281Updated this week
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆243Updated 5 months ago
- ☆71Updated 2 weeks ago
- An efficent implementation of the method proposed in "The Era of 1-bit LLMs"☆154Updated 8 months ago