liangyuwang / Tiny-MegatronLinks
Tiny-Megatron, a minimalistic re-implementation of the Megatron library
☆17Updated 2 months ago
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☆107Updated this week
- A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention☆221Updated 2 months ago
- analyse problems of AI with Math and Code☆27Updated 3 months ago
- ☆148Updated 4 months ago
- ☆43Updated last year
- Implement Flash Attention using Cute.☆96Updated 10 months ago
- The Official Implementation of Ada-KV [NeurIPS 2025]☆110Updated last month
- ☆44Updated this week
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆128Updated 2 weeks ago
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆48Updated 2 months ago
- Course materials for MIT6.5940: TinyML and Efficient Deep Learning Computing☆60Updated 10 months ago
- [DAC'25] Official implement of "HybriMoE: Hybrid CPU-GPU Scheduling and Cache Management for Efficient MoE Inference"☆87Updated 5 months ago
- [NeurIPS 2024] Efficient LLM Scheduling by Learning to Rank☆63Updated last year
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆48Updated last year
- Efficient Mixture of Experts for LLM Paper List☆143Updated last month
- ☆39Updated 3 months ago
- LLM Inference with Deep Learning Accelerator.☆53Updated 9 months ago
- ☆50Updated 5 months ago
- qwen-nsa☆83Updated last month
- 16-fold memory access reduction with nearly no loss☆106Updated 7 months ago
- DLSlime: Flexible & Efficient Heterogeneous Transfer Toolkit☆78Updated this week
- ☆149Updated 8 months ago
- ATC23 AE☆47Updated 2 years ago
- siiRL: Shanghai Innovation Institute RL Framework for Advanced LLMs and Multi-Agent Systems☆224Updated this week
- APRIL: Active Partial Rollouts in Reinforcement Learning to Tame Long-tail Generation. A system-level optimization for scalable LLM tra…☆41Updated last month
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆268Updated last week
- A sparse attention kernel supporting mix sparse patterns☆366Updated 9 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆152Updated last month
- ☆120Updated 2 months ago
- ☆31Updated 3 weeks ago