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
  ☆127Updated last year
• dongzhuoyao / uspace
  An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"
  ☆50Updated last year
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆254Updated 10 months ago
• Lakonik / GMFlow
  [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)
  ☆165Updated last month
• yinboc / dito
  Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"
  ☆159Updated 10 months ago
• NVlabs / DDO
  [ICML 2025 Spotlight] Direct Discriminative Optimization: Supercharging Diffusion/Autoregressive with GAN-type Discrimination
  ☆106Updated 4 months ago
• Jhomanik / Optimal-Flow-Matching
  The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)
  ☆100Updated 11 months ago
• tyuxie / RFM
  The official codebase for Reflected Flow Matching (ICML 2024)
  ☆21Updated last year
• leffff / FlowModels
  The aim of this repository is to test and implement Flow-Matching-based models
  ☆119Updated 11 months ago
• LINs-lab / UCGM
  [Preprint] UCGM: Unified Continuous Generative Models
  ☆170Updated 6 months ago
• CompVis / maskflow
  MaskFlow: Discrete Flows For Flexible and Efficient Long Video Generation
  ☆27Updated 9 months ago
• CompVis / tread
  ☆162Updated 2 months ago
• gstoica27 / DeltaFM
  [ICCV 2025] Official Implementation of Contrastive Flow Matching
  ☆150Updated 5 months ago
• mmathew23 / improved_edm
  Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"
  ☆95Updated last year
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆403Updated 2 weeks ago
• Neur-IO / OptVQ
  Towards training VQ-VAE models robustly!
  ☆87Updated 5 months ago
• HaoyiZhu / MeanFlow-PyTorch
  PyTorch re-implementation for MeanFlow
  ☆111Updated 4 months ago
• cfifty / rotation_trick
  ☆156Updated 8 months ago
• sangyun884 / fast-ode
  Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023
  ☆88Updated 10 months ago
• zh460045050 / VQGAN-LC
  ☆140Updated last year
• feizc / Dimba
  Transformer-Mamba Diffusion Models
  ☆117Updated last year
• mlvlab / CAF
  Official Implementation (Pytorch) of "Constant Acceleration Flow", NeurIPS 2024
  ☆34Updated last month
• CompVis / fm-boosting
  [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching
  ☆254Updated last month
• lucidrains / maskbit-pytorch
  Implementation of the proposed MaskBit from Bytedance AI
  ☆83Updated last year
• apple / ml-tarflow
  ☆305Updated 11 months ago
• lucidrains / multimodal-dit-pytorch
  Implementation of a multimodal diffusion transformer in Pytorch
  ☆107Updated last year
• Hhhhhhao / continuous_tokenizer
  ☆290Updated 6 months ago
• yhli123 / Immiscible-Diffusion
  Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment
  ☆61Updated 6 months ago
• thu-ml / DPM-Solver-v3
  Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)
  ☆130Updated last year
• Gsunshine / py-meanflow
  Pytorch implementation for MeanFlow
  ☆265Updated 4 months ago