liangyuwang / Tiny-MegatronLinks
Tiny-Megatron, a minimalistic re-implementation of the Megatron library
☆21Updated 5 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:
- A Survey of Efficient Attention Methods: Hardware-efficient, Sparse, Compact, and Linear Attention☆278Updated 2 months ago
- analyse problems of AI with Math and Code☆27Updated 6 months ago
- Tiny-DeepSpeed, a minimalistic re-implementation of the DeepSpeed library☆49Updated 5 months ago
- qwen-nsa☆87Updated 3 months ago
- Efficient Mixture of Experts for LLM Paper List☆166Updated 4 months ago
- DeepSeek Native Sparse Attention pytorch implementation☆114Updated last month
- ☆221Updated 2 months ago
- [ICLR 2025] PEARL: Parallel Speculative Decoding with Adaptive Draft Length☆147Updated last month
- [ASPLOS'26] Taming the Long-Tail: Efficient Reasoning RL Training with Adaptive Drafter☆130Updated 2 months ago
- The Official Implementation of Ada-KV [NeurIPS 2025]☆125Updated 2 months ago
- Pipeline-Parallel Lecture: Simplest Dualpipe Implementation.☆31Updated 4 months ago
- Implementation for FP8/INT8 Rollout for RL training without performence drop.☆288Updated 2 months ago
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆267Updated 6 months ago
- ☆152Updated 7 months ago
- Official PyTorch implementation of the paper "dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching" (dLLM-Cache…☆197Updated 2 months ago
- Code for paper: [ICLR2025 Oral] FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inference☆160Updated 3 months ago
- Accelerating Large-Scale Reasoning Model Inference with Sparse Self-Speculative Decoding☆87Updated 2 months ago
- [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification☆32Updated 10 months ago
- A curated list of recent papers on efficient video attention for video diffusion models, including sparsification, quantization, and cach…☆55Updated 3 months ago
- A lightweight Inference Engine built for block diffusion models☆40Updated last month
- ☆47Updated last year
- [ICML 2025 Oral] Mixture of Lookup Experts☆68Updated 2 months ago
- ☆129Updated 5 months ago
- 青稞Talk☆190Updated 2 weeks ago
- Fast and memory-efficient exact kmeans☆136Updated 2 months ago
- [NeurIPS'25 Spotlight] Adaptive Attention Sparsity with Hierarchical Top-p Pruning☆87Updated 2 months ago
- A sparse attention kernel supporting mix sparse patterns☆446Updated 2 weeks ago
- [Archived] For the latest updates and community contribution, please visit: https://github.com/Ascend/TransferQueue or https://gitcode.co…☆13Updated 3 weeks ago
- A lightweight reinforcement learning framework that integrates seamlessly into your codebase, empowering developers to focus on algorithm…☆99Updated 5 months ago
- ☆129Updated 7 months ago