Alternatives and similar repositories for nanoDiT

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

• bluorion-com / ZClip
  Official implementation of the paper: "ZClip: Adaptive Spike Mitigation for LLM Pre-Training".
  ☆141Updated last month
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆79Updated last year
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆107Updated last year
• lucidrains / maskbit-pytorch
  Implementation of the proposed MaskBit from Bytedance AI
  ☆83Updated last year
• SwayStar123 / microdiffusion
  ☆48Updated 10 months ago
• lucidrains / titok-pytorch
  Implementation of TiTok, proposed by Bytedance in "An Image is Worth 32 Tokens for Reconstruction and Generation"
  ☆183Updated last year
• ShangyuanTong / FreeFlow
  Official PyTorch Implementation of "Flow Map Distillation Without Data"
  ☆114Updated last month
• ArchiMickey / Just-a-DiT
  A repo of a modified version of Diffusion Transformer
  ☆45Updated 4 months ago
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆120Updated last year
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆139Updated last week
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆71Updated 2 months ago
• lucidrains / deep-cross-attention
  Implementation of the proposed DeepCrossAttention by Heddes et al at Google research, in Pytorch
  ☆96Updated 10 months ago
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆105Updated last year
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆129Updated 3 weeks ago
• huggingface / lora-fast
  Minimal repository to demonstrate fast LoRA inference with Flux family of models.
  ☆25Updated 5 months ago
• lucidrains / h-net-dynamic-chunking
  Implementation of the dynamic chunking mechanism in H-net by Hwang et al. of Carnegie Mellon
  ☆66Updated last week
• lucidrains / genie2-pytorch
  Implementation of a framework for Genie2 in Pytorch
  ☆156Updated last year
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆312Updated last year
• cloneofsimo / repa-rf
  ☆32Updated last year
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆90Updated last year
• cloneofsimo / min-max-in-dit
  ☆27Updated last year
• huggingface / flux-fast
  Making Flux go brrr on GPUs.
  ☆158Updated 2 weeks ago
• swookey-thinky / xdiffusion
  A unified media (Image, Video, Audio, Text) diffusion repository, for education and learning.
  ☆42Updated 9 months ago
• SwayStar123 / reimei
  ☆28Updated 3 months ago
• cloneofsimo / minDinoV2
  ☆23Updated last year
• sayakpaul / simple-image-recaptioning
  Recaption large (Web)Datasets with vllm and save the artifacts.
  ☆52Updated last year
• lucidrains / vit-arc-slot
  Explorations into improving ViTArc with Slot Attention
  ☆43Updated last year
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆126Updated last year
• CompVis / tread
  ☆167Updated last week
• OliverRensu / MVAR
  ☆71Updated last year