Alternatives and similar repositories for Just-a-DiT

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

• sangyun884 / rfpp
  The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024
  ☆123Updated last year
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆245Updated 9 months ago
• mmathew23 / improved_edm
  Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"
  ☆94Updated last year
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆114Updated 9 months ago
• yinboc / dito
  Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"
  ☆151Updated 8 months ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆385Updated 2 months ago
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆139Updated 4 months ago
• CompVis / fm-boosting
  [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching
  ☆246Updated this week
• sangyun884 / fast-ode
  Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023
  ☆88Updated 8 months ago
• VinAIResearch / LFM
  Official PyTorch implementation of the paper: Flow Matching in Latent Space
  ☆310Updated 8 months ago
• tyuxie / RFM
  The official codebase for Reflected Flow Matching (ICML 2024)
  ☆20Updated last year
• feizc / Dimba
  Transformer-Mamba Diffusion Models
  ☆116Updated last year
• dongzhuoyao / uspace
  An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"
  ☆46Updated last year
• thu-ml / DPM-Solver-v3
  Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)
  ☆125Updated last year
• locuslab / ect
  Consistency Models Made Easy
  ☆316Updated last year
• pkulwj1994 / diff_instruct
  official code for Diff-Instruct algorithm for one-step diffusion distillation
  ☆78Updated 9 months ago
• NVlabs / DDO
  [ICML 2025 Spotlight] Direct Discriminative Optimization: Supercharging Diffusion/Autoregressive with GAN-type Discrimination
  ☆102Updated 2 months ago
• sony / ctm
  ☆303Updated last year
• zh460045050 / VQGAN-LC
  ☆138Updated last year
• Jhomanik / Optimal-Flow-Matching
  The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)
  ☆94Updated 10 months ago
• White-Link / gpm
  Official implementation of the paper "Unifying GANs and Score-Based Diffusion as Generative Particle Models", NeurIPS 2023
  ☆26Updated 9 months ago
• Gsunshine / py-meanflow
  Pytorch implementation for MeanFlow
  ☆193Updated 2 months ago
• zelaki / eqvae
  [ICML'25] EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modeling.
  ☆154Updated 3 months ago
• tzco / Diffusion-wo-CFG
  Official Implementation for Diffusion Models Without Classifier-free Guidance
  ☆152Updated 8 months ago
• sail-sg / FDM
  The official PyTorch implementation of Fast Diffusion Model
  ☆95Updated 2 years ago
• feizc / DiS
  Scalable Diffusion Models with State Space Backbone
  ☆157Updated last year
• CompVis / discrete-interpolants
  The official implementation of "[MASK] is All You Need"
  ☆125Updated 2 months ago
• zhuyu-cs / MeanFlow
  Pytorch implementation of MeanFlow on ImageNet and CIFAR10
  ☆293Updated last month
• mingyuanzhou / SiD
  PyTorch code and model checkpoints for Score identity Distillation (SiD) and its adversarial version (SiDA)
  ☆136Updated 6 months ago
• mingyuanzhou / SiD-LSG
  Score identity Distillation with Long and Short Guidance for One-Step Text-to-Image Generation
  ☆89Updated 3 months ago