Alternatives and similar repositories for MINI-TORCH

Users that are interested in MINI-TORCH are comparing it to the libraries listed below

• xdit-project / DiTCacheAnalysis
  An auxiliary project analysis of the characteristics of KV in DiT Attention.
  ☆30Updated 6 months ago
• Dao-AILab / grouped-latent-attention
  ☆93Updated last week
• mit-han-lab / patch_conv
  Patch convolution to avoid large GPU memory usage of Conv2D
  ☆87Updated 4 months ago
• OpenNLPLab / LASP
  Linear Attention Sequence Parallelism (LASP)
  ☆83Updated last year
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆41Updated last month
• feifeibear / Odysseus-Transformer
  Odysseus: Playground of LLM Sequence Parallelism
  ☆70Updated 11 months ago
• Vchitect / LiteGen
  A light-weight and high-efficient training framework for accelerating diffusion tasks.
  ☆48Updated 8 months ago
• PipeFusion / PipeFusion
  A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters
  ☆45Updated 10 months ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆102Updated 11 months ago
• dame-cell / Triformer
  Transformers components but in Triton
  ☆33Updated 3 weeks ago
• nil0x9 / flash-muon
  Flash-Muon: An Efficient Implementation of Muon Optimizer
  ☆121Updated last week
• WaveSpeedAI / QuantumAttention
  [WIP] Better (FP8) attention for Hopper
  ☆30Updated 3 months ago
• timudk / flux_triton
  Writing FLUX in Triton
  ☆33Updated 8 months ago
• Juanerx / Q-DiT
  [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers
  ☆50Updated 9 months ago
• NVlabs / Fast-dLLM
  ☆144Updated this week
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆111Updated 5 months ago
• horseee / learning-to-cache
  [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching
  ☆104Updated 10 months ago
• lucidrains / mixture-of-attention
  Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts
  ☆119Updated 7 months ago
• thu-nics / DiTFastAttn
  ☆167Updated 4 months ago
• thu-ml / low-bit-optimizers
  Low-bit optimizers for PyTorch
  ☆128Updated last year
• TsinghuaC3I / Fourier-Position-Embedding
  [ICML 2025] Fourier Position Embedding: Enhancing Attention’s Periodic Extension for Length Generalization
  ☆70Updated this week
• frankxwang / dpo-prefix-sharing
  DPO, but faster 🚀
  ☆42Updated 6 months ago
• itsnamgyu / block-transformer
  Block Transformer: Global-to-Local Language Modeling for Fast Inference (NeurIPS 2024)
  ☆156Updated last month
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆70Updated last year
• BBuf / flash-rwkv
  ☆31Updated last year
• huggingface / amused
  ☆85Updated last year
• vedantroy / gpu_kernels
  ☆26Updated last year
• gpu-mode / ring-attention
  ring-attention experiments
  ☆145Updated 7 months ago
• tanyuqian / redco
  NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
  ☆66Updated 5 months ago
• Dao-AILab / gemm-cublas
  ☆20Updated last month