OpenMachine-ai / transformer-tricksLinks
A collection of tricks and tools to speed up transformer models
☆167Updated this week
Alternatives and similar repositories for transformer-tricks
Users that are interested in transformer-tricks are comparing it to the libraries listed below
Sorting:
- ☆56Updated 2 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆126Updated 6 months ago
- 🔥 A minimal training framework for scaling FLA models☆146Updated 3 weeks ago
- Efficient LLM Inference over Long Sequences☆376Updated this week
- Efficient triton implementation of Native Sparse Attention.☆155Updated 2 weeks ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆237Updated 4 months ago
- Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMs☆166Updated last week
- [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models☆221Updated last month
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆156Updated last month
- TransMLA: Multi-Head Latent Attention Is All You Need☆284Updated this week
- Work in progress.☆67Updated this week
- PyTorch implementation of models from the Zamba2 series.☆182Updated 4 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆121Updated last week
- ☆74Updated 3 months ago
- ☆47Updated 3 weeks ago
- Parallel Scaling Law for Language Model — Beyond Parameter and Inference Time Scaling☆367Updated 2 weeks ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆167Updated 2 months ago
- Fast and memory-efficient exact attention☆68Updated 3 months ago
- XAttention: Block Sparse Attention with Antidiagonal Scoring☆158Updated 3 weeks ago
- [ICML 2025] Fourier Position Embedding: Enhancing Attention’s Periodic Extension for Length Generalization☆70Updated this week
- Unofficial implementation for the paper "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆161Updated 11 months ago
- ☆129Updated 3 months ago
- The official implementation of the paper <MoA: Mixture of Sparse Attention for Automatic Large Language Model Compression>☆129Updated last week
- Q-GaLore: Quantized GaLore with INT4 Projection and Layer-Adaptive Low-Rank Gradients.☆199Updated 10 months ago
- [ICLR'25] Fast Inference of MoE Models with CPU-GPU Orchestration☆211Updated 6 months ago
- [ACL 2025 Main] EfficientQAT: Efficient Quantization-Aware Training for Large Language Models☆272Updated 2 weeks ago
- An efficent implementation of the method proposed in "The Era of 1-bit LLMs"☆153Updated 7 months ago
- Code accompanying the paper "Generalized Interpolating Discrete Diffusion"☆80Updated last week
- Linear Attention Sequence Parallelism (LASP)☆83Updated last year
- RWKV-7: Surpassing GPT☆88Updated 6 months ago