Alternatives and similar repositories for Just-a-DiT

Users that are interested in Just-a-DiT are comparing it to the libraries listed below

• dongzhuoyao / uspace
  An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"
  ☆42Updated last year
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆111Updated last month
• cloneofsimo / min-max-in-dit
  ☆27Updated last year
• mlvlab / CAF
  Official Implementation (Pytorch) of "Constant Acceleration Flow", NeurIPS 2024
  ☆34Updated 5 months ago
• CompVis / maskflow
  MaskFlow: Discrete Flows For Flexible and Efficient Long Video Generation
  ☆24Updated 4 months ago
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆103Updated 6 months ago
• zelaki / ReDi
  Boosting Generative Image Modeling via Joint Image-Feature Synthesis
  ☆47Updated last month
• NVlabs / DDO
  [ICML 2025 Spotlight] Direct Discriminative Optimization: Supercharging Diffusion/Autoregressive with GAN-type Discrimination
  ☆87Updated 3 weeks ago
• Jhomanik / Optimal-Flow-Matching
  The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)
  ☆78Updated 7 months ago
• jadehaus / preference-flow-matching
  Official code implementation for the work Preference Alignment with Flow Matching (NeurIPS 2024)
  ☆56Updated 8 months ago
• HaoyiZhu / MeanFlow-PyTorch
  PyTorch re-implementation for MeanFlow
  ☆69Updated 3 weeks ago
• sangyun884 / rfpp
  The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024
  ☆117Updated 9 months ago
• keishihara / flow-matching
  Flow Matching implemented in PyTorch
  ☆42Updated 6 months ago
• Zyriix / D2O
  Official implemention for Diffusion Models Are Innate One-Step Generators
  ☆23Updated 3 weeks ago
• LINs-lab / UCGM
  [Preprint] UCGM: Unified Continuous Generative Models
  ☆163Updated last month
• thunlp / ACDiT
  ACDiT: Interpolating Autoregressive Conditional Modeling and Diffusion Transformer
  ☆34Updated 6 months ago
• yhli123 / Immiscible-Diffusion
  Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment
  ☆56Updated last month
• taehong-moon / ee-diffusion
  Official Code Implementation for 'A Simple Early Exiting Framework for Accelerated Sampling in Diffusion Models'
  ☆19Updated 11 months ago
• lucidrains / maskbit-pytorch
  Implementation of the proposed MaskBit from Bytedance AI
  ☆82Updated 8 months ago
• DDCM-2025 / ddcm-compressed-image-generation
  Official implementation of the paper "Compressed Image Generation with Denoising Diffusion Codebook Models"
  ☆57Updated 5 months ago
• lucidrains / LVMAE-pytorch
  Implementation of the proposed LVMAE, from the paper, Extending Video Masked Autoencoders to 128 frames, in Pytorch
  ☆52Updated 7 months ago
• sayakpaul / nanoDiT
  Just another reasonably minimal repo for class-conditional training of pixel-space diffusion transformers.
  ☆114Updated last month
• jacklishufan / OmniFlows
  The official implementation of OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flows
  ☆76Updated last month
• mmathew23 / improved_edm
  Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"
  ☆95Updated last year
• facebookresearch / novel-view-acoustic-synthesis
  Code for Novel View Acoustic Synthesis paper
  ☆48Updated last year
• pkulwj1994 / diff_instruct
  official code for Diff-Instruct algorithm for one-step diffusion distillation
  ☆78Updated 6 months ago
• ali-vilab / FACM
  FACM: Flow-Anchored Consistency Models
  ☆78Updated last week
• CompVis / tread
  ☆56Updated this week
• cloneofsimo / minMomentMatching
  ☆11Updated 7 months ago
• yinboc / dito
  Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"
  ☆128Updated 5 months ago