OpenMachine-ai / transformer-tricksLinks
A collection of tricks and tools to speed up transformer models
☆193Updated last month
Alternatives and similar repositories for transformer-tricks
Users that are interested in transformer-tricks are comparing it to the libraries listed below
Sorting:
- mHC kernels implemented in CUDA☆217Updated this week
- ☆65Updated 9 months ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆225Updated 7 months ago
- Block Diffusion for Ultra-Fast Speculative Decoding☆313Updated last week
- The evaluation framework for training-free sparse attention in LLMs☆108Updated 3 months ago
- Efficient LLM Inference over Long Sequences☆393Updated 6 months ago
- ☆265Updated 7 months ago
- ☆63Updated 8 months ago
- Work in progress.☆77Updated last month
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆131Updated last year
- CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learning☆277Updated 2 months ago
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆421Updated 4 months ago
- ☆148Updated last year
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆163Updated 9 months ago
- FlashRNN - Fast RNN Kernels with I/O Awareness☆174Updated 2 months ago
- ☆103Updated 10 months ago
- Chain of Experts (CoE) enables communication between experts within Mixture-of-Experts (MoE) models☆226Updated 2 months ago
- Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMs☆198Updated last month
- Fused Qwen3 MoE layer for faster training, compatible with Transformers, LoRA, bnb 4-bit quant, Unsloth. Also possible to train LoRA over…☆225Updated this week
- Normalized Transformer (nGPT)☆195Updated last year
- RWKV-7: Surpassing GPT☆103Updated last year
- ☆133Updated 7 months ago
- Official repository for the paper "NeuZip: Memory-Efficient Training and Inference with Dynamic Compression of Neural Networks". This rep…☆60Updated last year
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆294Updated 7 months ago
- Training-free Post-training Efficient Sub-quadratic Complexity Attention. Implemented with OpenAI Triton.☆147Updated 2 months ago
- PyTorch implementation of models from the Zamba2 series.☆185Updated 11 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆249Updated 11 months ago
- Low-bit optimizers for PyTorch☆137Updated 2 years ago
- Pytorch Distributed native training library for LLMs/VLMs with OOTB Hugging Face support☆245Updated this week
- ☆61Updated 2 years ago