Alternatives and similar repositories for FSDP-Training

Users that are interested in FSDP-Training are comparing it to the libraries listed below

• ThisisBillhe / ZipCache
  [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification
  ☆32Updated 9 months ago
• horseee / dKV-Cache
  [NeurIPS'25] dKV-Cache: The Cache for Diffusion Language Models
  ☆127Updated 7 months ago
• mit-han-lab / x-attention
  [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆263Updated 6 months ago
• tilde-research / nsa-impl
  An efficient implementation of the NSA (Native Sparse Attention) kernel
  ☆128Updated 6 months ago
• OpenSparseLLMs / Linearization
  ☆62Updated 6 months ago
• ruikangliu / Quantized-Reasoning-Models
  [COLM 2025] Official PyTorch implementation of "Quantization Hurts Reasoning? An Empirical Study on Quantized Reasoning Models"
  ☆64Updated 6 months ago
• NoakLiu / FastCache-xDiT
  FastCache: Fast Caching for Diffusion Transformer Through Learnable Linear Approximation [Efficient ML Model]
  ☆45Updated 3 weeks ago
• Aaronhuang-778 / Mixture-Compressor-MoE
  [ICLR 2025] Mixture Compressor for Mixture-of-Experts LLMs Gains More
  ☆65Updated 10 months ago
• z-lab / sparselora
  [ICML 2025] SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity
  ☆64Updated 6 months ago
• xdit-project / DiTCacheAnalysis
  An auxiliary project analysis of the characteristics of KV in DiT Attention.
  ☆32Updated last year
• horseee / learning-to-cache
  [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching
  ☆116Updated last year
• Espere-1119-Song / VideoNSA
  VideoNSA: Native Sparse Attention Scales Video Understanding
  ☆76Updated last month
• JieShibo / MoLE
  [ICML 2025 Oral] Mixture of Lookup Experts
  ☆60Updated last month
• sramshetty / mixture-of-depths
  An unofficial implementation of "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"
  ☆36Updated last year
• thu-nics / MBQ
  The code repository of "MBQ: Modality-Balanced Quantization for Large Vision-Language Models"
  ☆72Updated 9 months ago
• fla-org / flash-bidirectional-linear-attention
  Triton implement of bi-directional (non-causal) linear attention
  ☆60Updated 11 months ago
• OpenMOSS / DiRL
  ☆118Updated last week
• maomaocun / dLLM-cache
  Official PyTorch implementation of the paper "dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching" (dLLM-Cache…
  ☆191Updated last month
• chuanyang-Zheng / DAPE
  The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"
  ☆40Updated last year
• Juanerx / Q-DiT
  [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers
  ☆73Updated last year
• lliai / D2MoE
  D^2-MoE: Delta Decompression for MoE-based LLMs Compression
  ☆72Updated 9 months ago
• svg-project / flash-kmeans
  Fast and memory-efficient exact kmeans
  ☆131Updated last month
• OpenSparseLLMs / MoM
  ☆114Updated 3 months ago
• ThisisBillhe / ZipAR
  [ICML 2025] This is the official PyTorch implementation of "ZipAR: Accelerating Auto-regressive Image Generation through Spatial Locality…
  ☆53Updated 9 months ago
• StargazerX0 / ScaleKV
  [NeurIPS 2025] ScaleKV: Memory-Efficient Visual Autoregressive Modeling with Scale-Aware KV Cache Compression
  ☆50Updated 2 months ago
• inclusionAI / dFactory
  Easy and Efficient dLLM Fine-Tuning
  ☆190Updated 3 weeks ago
• zhijie-group / Discrete-Diffusion-Forcing
  Discrete Diffusion Forcing (D2F): dLLMs Can Do Faster-Than-AR Inference
  ☆224Updated 3 months ago
• ThisisBillhe / torch_quantizer
  torch_quantizer is a out-of-box quantization tool for PyTorch models on CUDA backend, specially optimized for Diffusion Models.
  ☆22Updated last year
• SUSTechBruce / LOOK-M
  [EMNLP 2024 Findings🔥] Official implementation of ": LOOK-M: Look-Once Optimization in KV Cache for Efficient Multimodal Long-Context In…
  ☆103Updated last year
• zhijie-group / AdaMoE
  [Findings of EMNLP 2024] AdaMoE: Token-Adaptive Routing with Null Experts for Mixture-of-Experts Language Models
  ☆17Updated last year