Triton-based implementation of Sparse Mixture of Experts.
☆280Oct 3, 2025Updated 11 months ago
Alternatives and similar repositories for scattermoe
Users that are interested in scattermoe are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆112Aug 26, 2024Updated 2 years ago
- sigma-MoE layer☆21Jan 5, 2024Updated 2 years ago
- Applied AI experiments and examples for PyTorch☆323Aug 22, 2025Updated last year
- extensible collectives library in triton☆97Mar 31, 2025Updated last year
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆604Aug 14, 2026Updated 2 weeks ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Transformers components but in Triton☆34May 9, 2025Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆154May 29, 2025Updated last year
- LM engine is a library for pretraining/finetuning LLMs☆190Aug 10, 2026Updated 3 weeks ago
- ☆1,589Mar 25, 2026Updated 5 months ago
- Ring attention implementation with flash attention☆1,050Sep 10, 2025Updated 11 months ago
- ☆93Aug 18, 2024Updated 2 years ago
- Fast low-bit matmul kernels in Triton☆483Aug 9, 2026Updated 3 weeks ago
- [MLSys'25] QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving; [MLSys'25] LServe: Efficient Long-sequence LLM Se…☆855Mar 6, 2025Updated last year
- ☆272Jul 11, 2024Updated 2 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Tile primitives for speedy kernels☆3,664Aug 28, 2026Updated last week
- Odysseus: Playground of LLM Sequence Parallelism☆83Jun 17, 2024Updated 2 years ago
- A Quirky Assortment of CuTe Kernels☆1,136Updated this week
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆347Jul 2, 2024Updated 2 years ago
- ModuleFormer is a MoE-based architecture that includes two different types of experts: stick-breaking attention heads and feedforward exp…☆225Sep 18, 2025Updated 11 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆192Apr 8, 2026Updated 4 months ago
- ☆20May 30, 2024Updated 2 years ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆367Updated this week
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆101Sep 30, 2024Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Aug 19, 2024Updated 2 years ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆75Aug 2, 2024Updated 2 years ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆39Jun 11, 2025Updated last year
- Patch convolution to avoid large GPU memory usage of Conv2D☆95Jan 23, 2025Updated last year
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆157Dec 4, 2024Updated last year
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,358Aug 28, 2025Updated last year
- Distributed Compiler and Optimized Parallel Kernels☆1,532Aug 12, 2026Updated 3 weeks ago
- A throughput-oriented high-performance serving framework for LLMs☆974Mar 29, 2026Updated 5 months ago
- ☆213Jun 17, 2026Updated 2 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Tutel MoE: Optimized Mixture-of-Experts Library, Support GptOss/DeepSeek/Kimi-K2/Qwen3 using FP8/NVFP4/MXFP4☆1,016Updated this week
- A Easy-to-understand TensorOp Matmul Tutorial☆452Mar 5, 2026Updated 5 months ago
- ☆172Dec 27, 2024Updated last year
- A collection of memory efficient attention operators implemented in the Triton language.☆307Updated this week
- Experiment of using Tangent to autodiff triton☆81Jan 22, 2024Updated 2 years ago
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,473Updated this week
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆135Jun 24, 2025Updated last year