liangyuwang / Tiny-MegatronLinks
Tiny-Megatron, a minimalistic re-implementation of the Megatron library
☆15Updated this week
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☆83Updated 5 months ago
- ☆42Updated last year
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆102Updated 3 months ago
- [NeurIPS 2024] Efficient LLM Scheduling by Learning to Rank☆51Updated 9 months ago
- [DAC'25] Official implement of "HybriMoE: Hybrid CPU-GPU Scheduling and Cache Management for Efficient MoE Inference"☆64Updated last month
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆38Updated last week
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆48Updated last year
- analyse problems of AI with Math and Code☆18Updated last week
- siiRL: Shanghai Innovation Institute RL Framework for Advanced LLMs and Multi-Agent Systems☆152Updated this week
- ☆50Updated 2 months ago
- 16-fold memory access reduction with nearly no loss☆103Updated 4 months ago
- ☆78Updated 3 months ago
- 青稞社区☆26Updated this week
- Official implementation of "Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decoding"☆320Updated this week
- qwen-nsa☆70Updated 3 months ago
- ☆38Updated 3 months ago
- A sparse attention kernel supporting mix sparse patterns☆262Updated 5 months ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆40Updated last month
- Accelerate LLM preference tuning via prefix sharing with a single line of code☆42Updated last month
- ☆145Updated 5 months ago
- ATC23 AE☆46Updated 2 years ago
- The Official Implementation of Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inference☆87Updated last month
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆110Updated last year
- Implement some method of LLM KV Cache Sparsity☆34Updated last year
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆124Updated 2 months ago
- LLM Inference with Deep Learning Accelerator.☆49Updated 6 months ago
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆213Updated 3 weeks ago
- ☆123Updated 2 months ago
- Bridge Megatron-Core to Hugging Face/Reinforcement Learning☆67Updated this week
- Course materials for MIT6.5940: TinyML and Efficient Deep Learning Computing☆51Updated 6 months ago