Alternatives and similar repositories for nanoDiT

Users that are interested in nanoDiT are comparing it to the libraries listed below

• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆74Updated 6 months ago
• SwayStar123 / microdiffusion
  ☆47Updated 4 months ago
• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆127Updated 3 weeks ago
• cloneofsimo / repa-rf
  ☆33Updated 7 months ago
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆97Updated 7 months ago
• sayakpaul / simple-image-recaptioning
  Recaption large (Web)Datasets with vllm and save the artifacts.
  ☆52Updated 7 months ago
• cloneofsimo / efae
  ☆23Updated last year
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆64Updated 3 weeks ago
• lucidrains / maskbit-pytorch
  Implementation of the proposed MaskBit from Bytedance AI
  ☆82Updated 7 months ago
• AnonymousAlethiometer / SGD_SaI
  Official PyTorch Implementation for Paper "No More Adam: Learning Rate Scaling at Initialization is All You Need"
  ☆52Updated 4 months ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆102Updated last year
• cloneofsimo / infinite-fractal-stream
  ☆30Updated 8 months ago
• fal-ai-community / nano-mdm
  Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun
  ☆52Updated 3 months ago
• SwayStar123 / reimei
  ☆24Updated last month
• cloneofsimo / min-max-in-dit
  ☆27Updated last year
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆85Updated last week
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆173Updated last year
• cloneofsimo / imagenet.int8
  ☆39Updated last year
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆281Updated 8 months ago
• alexanderswerdlow / unidisc
  UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…
  ☆107Updated 2 months ago
• OliverRensu / MVAR
  ☆70Updated 7 months ago
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆105Updated 3 weeks ago
• rhfeiyang / art-free-diffusion
  Official implementation of "Art-Free Generative Models: Art Creation Without Graphic Art Knowledge"
  ☆31Updated 2 months ago
• cloneofsimo / minDinoV2
  ☆20Updated 8 months ago
• Gengzigang / TokenSet
  Official PyTorch implementation of TokenSet.
  ☆121Updated 3 months ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆122Updated 10 months ago
• lucidrains / deep-cross-attention
  Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch
  ☆88Updated 4 months ago
• poloclub / ClickDiffusion
  ClickDiffusion: Harnessing LLMs for Interactive Precise Image Editing
  ☆69Updated last year
• fal-ai-community / NativeSparseAttention
  research impl of Native Sparse Attention (2502.11089)
  ☆54Updated 4 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆63Updated 5 months ago