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
  ☆75Updated 11 months ago
• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆139Updated 2 weeks ago
• lucidrains / maskbit-pytorch
  Implementation of the proposed MaskBit from Bytedance AI
  ☆83Updated last year
• SwayStar123 / microdiffusion
  ☆49Updated 9 months ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆107Updated 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
  ☆68Updated 3 weeks ago
• 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
  ☆93Updated 5 months ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆116Updated last month
• sayakpaul / simple-image-recaptioning
  Recaption large (Web)Datasets with vllm and save the artifacts.
  ☆52Updated last year
• lucidrains / h-net-dynamic-chunking
  Implementation of the dynamic chunking mechanism in H-net by Hwang et al. of Carnegie Mellon
  ☆65Updated 3 months ago
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆105Updated last year
• ShangyuanTong / FreeFlow
  Official PyTorch Implementation of "Flow Map Distillation Without Data"
  ☆82Updated 2 weeks ago
• AnonymousAlethiometer / SGD_SaI
  Official PyTorch Implementation for Paper "No More Adam: Learning Rate Scaling at Initialization is All You Need"
  ☆54Updated 10 months ago
• huggingface / flux-fast
  Making Flux go brrr on GPUs.
  ☆157Updated 4 months ago
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆181Updated last year
• fal-ai-community / nano-mdm
  Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun
  ☆57Updated 8 months ago
• cloneofsimo / repa-rf
  ☆32Updated last year
• lucidrains / adam-atan2-pytorch
  Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch
  ☆134Updated last month
• cloneofsimo / efae
  ☆23Updated last year
• Gengzigang / TokenSet
  Official PyTorch implementation of TokenSet.
  ☆127Updated 8 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆306Updated last year
• cloneofsimo / minDinoV2
  ☆22Updated last year
• lucidrains / deep-cross-attention
  Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch
  ☆95Updated 9 months ago
• ArchiMickey / Just-a-DiT
  A repo of a modified version of Diffusion Transformer
  ☆42Updated 2 months ago
• huggingface / fineVideo
  ☆93Updated last year
• cloneofsimo / imagenet.int8
  ☆39Updated last year
• HanGuo97 / log-linear-attention
  ☆256Updated 6 months ago
• NVlabs / GSPN
  [CVPR 2025] Parallel Sequence Modeling via Generalized Spatial Propagation Network
  ☆108Updated 4 months ago
• deepreinforce-ai / CUDA-L1
  CUDA-L1: Improving CUDA Optimization via Contrastive Reinforcement Learning
  ☆247Updated last month