Alternatives and similar repositories for JiT

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

• hustvl / LightningDiT
  [CVPR 2025 Oral] Reconstruction vs. Generation: Taming Optimization Dilemma in Latent Diffusion Models
  ☆1,388Updated last month
• haidog-yaqub / MeanFlow
  Pytorch Implementation (unofficial) of the paper "Mean Flows for One-step Generative Modeling" by Geng et al.
  ☆1,055Updated last month
• bytetriper / RAE
  Official PyTorch Implementation of "Diffusion Transformers with Representation Autoencoders"
  ☆1,746Updated 3 weeks ago
• LTH14 / fractalgen
  PyTorch implementation of FractalGen https://arxiv.org/abs/2502.17437
  ☆1,220Updated 11 months ago
• sihyun-yu / REPA
  [ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think
  ☆1,534Updated 10 months ago
• Gsunshine / meanflow
  JAX implementation of MeanFlow
  ☆539Updated last week
• zhuyu-cs / MeanFlow
  Pytorch implementation of MeanFlow on ImageNet and CIFAR10
  ☆421Updated last month
• LTH14 / mar
  PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838
  ☆1,857Updated last year
• kvfrans / shortcut-models
  ☆735Updated last year
• ChaofanTao / Autoregressive-Models-in-Vision-Survey
  [TMLR 2025🔥] A survey for the autoregressive models in vision.
  ☆787Updated 3 months ago
• brownvc / R3GAN
  Code for NeurIPS 2024 paper - The GAN is dead; long live the GAN! A Modern Baseline GAN - by Huang et al.
  ☆853Updated last year
• bytedance / 1d-tokenizer
  This repo contains the code for 1D tokenizer and generator
  ☆1,109Updated 10 months ago
• TongTong313 / rectified-flow
  从零手搓Flow Matching（Rectified Flow）
  ☆593Updated 2 months ago
• gnobitab / RectifiedFlow
  Official Implementation of Rectified Flow (ICLR2023 Spotlight)
  ☆1,543Updated last year
• Gsunshine / py-meanflow
  Pytorch implementation for MeanFlow
  ☆313Updated 6 months ago
• yifan123 / flow_grpo
  [NeurIPS 2025] An official implementation of Flow-GRPO: Training Flow Matching Models via Online RL
  ☆1,979Updated 3 months ago
• lxa9867 / Awesome-Autoregressive-Visual-Generation
  This is a repo to track the latest autoregressive visual generation papers.
  ☆429Updated 7 months ago
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆432Updated 2 months ago
• willisma / SiT
  Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"
  ☆1,092Updated last month
• NVlabs / DiffiT
  [ECCV 2024] Official Repository for DiffiT: Diffusion Vision Transformers for Image Generation
  ☆506Updated last year
• lucidrains / transfusion-pytorch
  Pytorch implementation of Transfusion, "Predict the Next Token and Diffuse Images with One Multi-Modal Model", from MetaAI
  ☆1,322Updated 2 weeks ago
• Hhhhhhao / continuous_tokenizer
  ☆304Updated 8 months ago
• sail-sg / MDT
  Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
  ☆594Updated last year
• End2End-Diffusion / REPA-E
  [ICCV 2025] Official implementation of the paper: REPA-E: Unlocking VAE for End-to-End Tuning of Latent Diffusion Transformers
  ☆449Updated 2 months ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆816Updated last year
• LTH14 / rcg
  PyTorch implementation of RCG https://arxiv.org/abs/2312.03701
  ☆936Updated last year
• JunyaoHu / common_metrics_on_video_quality
  You can easily calculate FVD, PSNR, SSIM, LPIPS for evaluating the quality of generated or predicted videos.
  ☆546Updated 3 weeks ago
• lxa9867 / ImageFolder
  High-performance Image Tokenizers for VAR and AR
  ☆302Updated 9 months ago
• CompVis / zigma
  A PyTorch implementation of the paper "ZigMa: A DiT-Style Mamba-based Diffusion Model" (ECCV 2024)
  ☆343Updated 10 months ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆424Updated last week