Triton-based implementation of Sparse Mixture of Experts.
☆281Oct 3, 2025Updated 9 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:
- ☆113Aug 26, 2024Updated last year
- sigma-MoE layer☆21Jan 5, 2024Updated 2 years ago
- Applied AI experiments and examples for PyTorch☆321Aug 22, 2025Updated 11 months ago
- 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.☆605May 13, 2026Updated 2 months 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☆184Updated this week
- ☆1,583Mar 25, 2026Updated 4 months ago
- Ring attention implementation with flash attention☆1,037Sep 10, 2025Updated 10 months ago
- ☆93Aug 18, 2024Updated last year
- Fast low-bit matmul kernels in Triton☆477Jul 15, 2026Updated last week
- [MLSys'25] QServe: W4A8KV4 Quantization and System Co-design for Efficient LLM Serving; [MLSys'25] LServe: Efficient Long-sequence LLM Se…☆850Mar 6, 2025Updated last year
- ☆270Jul 11, 2024Updated 2 years 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.
- Tile primitives for speedy kernels☆3,563Jul 13, 2026Updated last week
- Odysseus: Playground of LLM Sequence Parallelism☆83Jun 17, 2024Updated 2 years ago
- A Quirky Assortment of CuTe Kernels☆1,070Updated this week
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆344Jul 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 10 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆192Apr 8, 2026Updated 3 months ago
- ☆20May 30, 2024Updated 2 years ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆362Updated this week
- Official repository for the paper "SwitchHead: Accelerating Transformers with Mixture-of-Experts Attention"☆101Sep 30, 2024Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆223Aug 19, 2024Updated last year
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆75Aug 2, 2024Updated last year
- 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☆155Dec 4, 2024Updated last year
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,345Aug 28, 2025Updated 10 months ago
- Distributed Compiler based on Triton for Parallel Systems☆1,498Updated this week
- A throughput-oriented high-performance serving framework for LLMs☆969Mar 29, 2026Updated 3 months ago
- ☆212Jun 17, 2026Updated last month
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Tutel MoE: Optimized Mixture-of-Experts Library, Support GptOss/DeepSeek/Kimi-K2/Qwen3 using FP8/NVFP4/MXFP4☆1,000Updated this week
- A Easy-to-understand TensorOp Matmul Tutorial☆445Mar 5, 2026Updated 4 months ago
- ☆170Dec 27, 2024Updated last year
- A collection of memory efficient attention operators implemented in the Triton language.☆301Updated this week
- Experiment of using Tangent to autodiff triton☆81Jan 22, 2024Updated 2 years ago
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,390Updated this week
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆133Jun 24, 2025Updated last year