microsoft / SparseMixerLinks
Sparse Backpropagation for Mixture-of-Expert Training
☆29Updated last year
Alternatives and similar repositories for SparseMixer
Users that are interested in SparseMixer are comparing it to the libraries listed below
Sorting:
- Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"☆121Updated last year
- ☆57Updated last year
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆29Updated last year
- Long Context Extension and Generalization in LLMs☆62Updated last year
- ☆106Updated last year
- Using FlexAttention to compute attention with different masking patterns☆47Updated last year
- Stick-breaking attention☆61Updated 5 months ago
- ☆56Updated 6 months ago
- Language models scale reliably with over-training and on downstream tasks☆100Updated last year
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆105Updated 2 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆73Updated last year
- Kinetics: Rethinking Test-Time Scaling Laws☆84Updated 5 months ago
- ☆35Updated last year
- ☆50Updated last year
- ☆106Updated last year
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆56Updated 2 years ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆86Updated last year
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆40Updated last month
- Simple and efficient pytorch-native transformer training and inference (batched)☆79Updated last year
- 32 times longer context window than vanilla Transformers and up to 4 times longer than memory efficient Transformers.☆49Updated 2 years ago
- ☆31Updated 2 years ago
- ☆20Updated 3 years ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- Here we will test various linear attention designs.☆62Updated last year
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆60Updated last year
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆81Updated 2 years ago
- Official code repo for paper "Great Memory, Shallow Reasoning: Limits of kNN-LMs"☆24Updated 7 months ago
- Code for the ACL-2022 paper "StableMoE: Stable Routing Strategy for Mixture of Experts"☆51Updated 3 years ago
- Muon fsdp 2☆46Updated 4 months ago
- This package implements THOR: Transformer with Stochastic Experts.☆65Updated 4 years ago