Alternatives and similar repositories for FSDP-Training

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

• mit-han-lab / x-attention
  [ICML 2025] XAttention: Block Sparse Attention with Antidiagonal Scoring
  ☆267Updated 6 months ago
• horseee / dKV-Cache
  [NeurIPS'25] dKV-Cache: The Cache for Diffusion Language Models
  ☆129Updated 8 months ago
• z-lab / sparselora
  [ICML 2025] SparseLoRA: Accelerating LLM Fine-Tuning with Contextual Sparsity
  ☆69Updated 6 months ago
• ThisisBillhe / ZipCache
  [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification
  ☆32Updated 10 months ago
• thu-nics / R2R
  [NeurIPS'25] The official code implementation for paper "R2R: Efficiently Navigating Divergent Reasoning Paths with Small-Large Model Tok…
  ☆75Updated this week
• xdit-project / DiTCacheAnalysis
  An auxiliary project analysis of the characteristics of KV in DiT Attention.
  ☆32Updated last year
• OpenSparseLLMs / Linearization
  ☆64Updated 6 months ago
• fla-org / flash-bidirectional-linear-attention
  Triton implement of bi-directional (non-causal) linear attention
  ☆64Updated 11 months ago
• ruikangliu / Quantized-Reasoning-Models
  [COLM 2025] Official PyTorch implementation of "Quantization Hurts Reasoning? An Empirical Study on Quantized Reasoning Models"
  ☆67Updated 6 months ago
• tilde-research / nsa-impl
  An efficient implementation of the NSA (Native Sparse Attention) kernel
  ☆128Updated 7 months ago
• ThisisBillhe / ZipAR
  [ICML 2025] This is the official PyTorch implementation of "ZipAR: Accelerating Auto-regressive Image Generation through Spatial Locality…
  ☆53Updated 10 months ago
• Aaronhuang-778 / Mixture-Compressor-MoE
  [ICLR 2025] Mixture Compressor for Mixture-of-Experts LLMs Gains More
  ☆66Updated 11 months ago
• sramshetty / mixture-of-depths
  An unofficial implementation of "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"
  ☆36Updated last year
• NoakLiu / FastCache-xDiT
  FastCache: Fast Caching for Diffusion Transformer Through Learnable Linear Approximation [Efficient ML Model]
  ☆46Updated this week
• Juanerx / Q-DiT
  [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers
  ☆74Updated last year
• 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
• thu-nics / MBQ
  The code repository of "MBQ: Modality-Balanced Quantization for Large Vision-Language Models"
  ☆75Updated 10 months ago
• htqin / IR-QLoRA
  [ICML 2024 Oral] This project is the official implementation of our Accurate LoRA-Finetuning Quantization of LLMs via Information Retenti…
  ☆67Updated last year
• JieShibo / MoLE
  [ICML 2025 Oral] Mixture of Lookup Experts
  ☆68Updated last month
• maomaocun / dLLM-cache
  Official PyTorch implementation of the paper "dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching" (dLLM-Cache…
  ☆197Updated 2 months ago
• 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
  ☆79Updated 2 months ago
• svg-project / flash-kmeans
  Fast and memory-efficient exact kmeans
  ☆136Updated 2 months ago
• pprp / Pruner-Zero
  [ICML24] Pruner-Zero: Evolving Symbolic Pruning Metric from scratch for LLMs
  ☆98Updated last year
• OpenMOSS / DiRL
  ☆137Updated last week
• chuanyang-Zheng / DAPE
  The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"
  ☆40Updated last year
• LiangrunFlora / Slow-Fast-Sampling
  Official PyTorch implementation of the paper "Accelerating Diffusion Large Language Models with SlowFast Sampling: The Three Golden Princ…
  ☆39Updated 6 months ago
• ModelTC / QLLM
  [ICLR 2024] This is the official PyTorch implementation of "QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Mod…
  ☆39Updated last year
• A-suozhang / ViDiT-Q
  ☆15Updated 10 months ago
• lliai / D2MoE
  D^2-MoE: Delta Decompression for MoE-based LLMs Compression
  ☆72Updated 10 months ago