LinB203 / FSDP-TrainingLinks
Minimal PyTorch implementation of TP, SP, FSDP and sharded-EMA
☆29Updated 2 weeks ago
Alternatives and similar repositories for FSDP-Training
Users that are interested in FSDP-Training are comparing it to the libraries listed below
Sorting:
- [ICML 2025] SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity☆64Updated 5 months ago
- [ICLR 2025] Mixture Compressor for Mixture-of-Experts LLMs Gains More☆63Updated 10 months ago
- [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring☆256Updated 5 months ago
- [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers☆73Updated last year
- [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification☆30Updated 8 months ago
- [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching☆116Updated last year
- ☆93Updated last week
- [NeurIPS'25] dKV-Cache: The Cache for Diffusion Language Models☆123Updated 6 months ago
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆32Updated last year
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆126Updated 5 months ago
- An unofficial implementation of "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆36Updated last year
- Official PyTorch implementation of the paper "dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching" (dLLM-Cache…☆186Updated last month
- [COLM 2025] Official PyTorch implementation of "Quantization Hurts Reasoning? An Empirical Study on Quantized Reasoning Models"☆61Updated 5 months ago
- [ICML 2025 Oral] Mixture of Lookup Experts☆56Updated 2 weeks ago
- The code repository of "MBQ: Modality-Balanced Quantization for Large Vision-Language Models"☆70Updated 9 months ago
- [ICML 2025] This is the official PyTorch implementation of "ZipAR: Accelerating Auto-regressive Image Generation through Spatial Locality…☆53Updated 8 months ago
- FastCache: Fast Caching for Diffusion Transformer Through Learnable Linear Approximation [Efficient ML Model]☆45Updated this week
- ☆187Updated 11 months ago
- [ICLR25] STBLLM: Breaking the 1-Bit Barrier with Structured Binary LLMs☆16Updated 6 months ago
- ☆62Updated 5 months ago
- Locality-aware Parallel Decoding for Efficient Autoregressive Image Generation☆80Updated 5 months ago
- [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…☆66Updated last year
- [ICML 2024 Oral] This project is the official implementation of our Accurate LoRA-Finetuning Quantization of LLMs via Information Retenti…☆67Updated last year
- VideoNSA: Native Sparse Attention Scales Video Understanding☆70Updated last month
- [NeurIPS'25] The official code implementation for paper "R2R: Efficiently Navigating Divergent Reasoning Paths with Small-Large Model Tok…☆67Updated this week
- FORA introduces simple yet effective caching mechanism in Diffusion Transformer Architecture for faster inference sampling.☆52Updated last year
- Official repository for the paper Local Linear Attention: An Optimal Interpolation of Linear and Softmax Attention For Test-Time Regressi…☆23Updated 2 months ago
- SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention☆146Updated last month
- ☆65Updated 7 months ago
- Triton implement of bi-directional (non-causal) linear attention☆58Updated 10 months ago