Dao-AILab / sonic-moeLinks
Accelerating MoE with IO and Tile-aware Optimizations
☆563Updated last week
Alternatives and similar repositories for sonic-moe
Users that are interested in sonic-moe are comparing it to the libraries listed below
Sorting:
- A Quirky Assortment of CuTe Kernels☆772Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆319Updated this week
- Autonomous GPU Kernel Generation via Deep Agents☆223Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆141Updated 8 months ago
- Applied AI experiments and examples for PyTorch☆314Updated 5 months ago
- Fast low-bit matmul kernels in Triton☆424Updated this week
- kernels, of the mega variety☆657Updated 4 months ago
- Distributed MoE in a Single Kernel [NeurIPS '25]☆188Updated this week
- ☆277Updated this week
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆264Updated last month
- Allow torch tensor memory to be released and resumed later☆213Updated 2 weeks ago
- Cataloging released Triton kernels.☆291Updated 4 months ago
- [ICML 2024] Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inference☆368Updated 6 months ago
- ☆342Updated this week
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆163Updated 2 months ago
- Zero Bubble Pipeline Parallelism☆449Updated 8 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆263Updated 3 months ago
- Perplexity GPU Kernels☆554Updated 2 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆161Updated 3 months ago
- ☆158Updated last year
- Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.☆158Updated last week
- Pipeline Parallelism Emulation and Visualization☆76Updated 3 weeks ago
- Utility scripts for PyTorch (e.g. Make Perfetto show some disappearing kernels, Memory profiler that understands more low-level allocatio…☆82Updated 4 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆335Updated last year
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆276Updated 6 months ago
- JAX backend for SGL☆232Updated this week
- Dynamic Memory Management for Serving LLMs without PagedAttention☆457Updated 8 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Updated last year
- Building the Virtuous Cycle for AI-driven LLM Systems☆140Updated this week
- Helpful kernel tutorials and examples for tile-based GPU programming☆617Updated this week