liangyuwang / Tiny-MegatronLinks
Tiny-Megatron, a minimalistic re-implementation of the Megatron library
☆16Updated last month
Alternatives and similar repositories for Tiny-Megatron
Users that are interested in Tiny-Megatron are comparing it to the libraries listed below
Sorting:
- DeepSeek Native Sparse Attention pytorch implementation☆106Updated 2 weeks ago
- analyse problems of AI with Math and Code☆26Updated 2 months ago
- A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention☆200Updated last month
- ☆43Updated last year
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆120Updated 6 months ago
- [DAC'25] Official implement of "HybriMoE: Hybrid CPU-GPU Scheduling and Cache Management for Efficient MoE Inference"☆75Updated 4 months ago
- Implement Flash Attention using Cute.☆96Updated 10 months ago
- LLM Inference with Deep Learning Accelerator.☆52Updated 9 months ago
- ☆39Updated 2 months ago
- qwen-nsa☆79Updated last week
- ☆148Updated 3 months ago
- ☆148Updated 7 months ago
- DLSlime: Flexible & Efficient Heterogeneous Transfer Toolkit☆70Updated this week
- ☆50Updated 5 months ago
- ☆120Updated 2 months ago
- A sparse attention kernel supporting mix sparse patterns☆322Updated 8 months ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆45Updated 4 months ago
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆48Updated last year
- siiRL: Shanghai Innovation Institute RL Framework for Advanced LLMs and Multi-Agent Systems☆222Updated this week
- ☆65Updated 5 months ago
- 16-fold memory access reduction with nearly no loss☆105Updated 6 months ago
- [NeurIPS 2024] Efficient LLM Scheduling by Learning to Rank☆60Updated 11 months ago
- 注释的nano_vllm仓库,并且完成了MiniCPM4的适配以及注册新模型的功能☆81Updated 2 months ago
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆60Updated 2 weeks ago
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆47Updated 2 months ago
- Implement some method of LLM KV Cache Sparsity☆39Updated last year
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆121Updated 5 months ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆115Updated last year
- The Official Implementation of Ada-KV [NeurIPS 2025]☆106Updated last month
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆147Updated last week