OpenMachine-ai / transformer-tricksLinks
A collection of tricks and tools to speed up transformer models
☆194Updated 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☆249Updated 3 weeks ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆233Updated 7 months ago
- ☆270Updated 8 months ago
- RWKV-7: Surpassing GPT☆104Updated last year
- ☆67Updated 10 months ago
- ☆131Updated 8 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆131Updated last year
- Work in progress.☆79Updated 2 months ago
- ☆105Updated 11 months ago
- The evaluation framework for training-free sparse attention in LLMs☆117Updated last week
- Efficient LLM Inference over Long Sequences☆394Updated 7 months ago
- ☆64Updated 8 months ago
- Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)☆163Updated 9 months ago
- Fused Qwen3 MoE layer for faster training, compatible with Transformers, LoRA, bnb 4-bit quant, Unsloth. Also possible to train LoRA over…☆231Updated this week
- Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI☆293Updated 8 months ago
- Memory optimized Mixture of Experts☆73Updated 6 months ago
- Normalized Transformer (nGPT)☆198Updated last year
- Chain of Experts (CoE) enables communication between experts within Mixture-of-Experts (MoE) models☆228Updated 3 months ago
- Towards Economical Inference: Enabling DeepSeek's Multi-Head Latent Attention in Any Transformer-based LLMs☆204Updated 2 months ago
- Block Diffusion for Ultra-Fast Speculative Decoding☆459Updated this week
- ☆148Updated last year
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆252Updated last year
- [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule☆446Updated 4 months ago
- ☆71Updated 7 months ago
- Official implementation for Training LLMs with MXFP4☆118Updated 9 months ago
- Official repository for the paper "NeuZip: Memory-Efficient Training and Inference with Dynamic Compression of Neural Networks". This rep…☆60Updated last year
- QeRL enables RL for 32B LLMs on a single H100 GPU.☆481Updated 2 months ago
- implementations and experimentation on mHC by deepseek - https://arxiv.org/abs/2512.24880☆290Updated this week
- PyTorch implementation of models from the Zamba2 series.☆186Updated last year
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆128Updated 7 months ago