ArchiMickey / Just-a-DiTLinks
A repo of a modified version of Diffusion Transformer
☆32Updated last year
Alternatives and similar repositories for Just-a-DiT
Users that are interested in Just-a-DiT are comparing it to the libraries listed below
Sorting:
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆119Updated 9 months ago
- The aim of this repository is to test and implement Flow-Matching-based models☆103Updated 6 months ago
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆123Updated 2 months ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆95Updated last year
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆235Updated 6 months ago
- Official code for "DPM-Solver-v3: Improved Diffusion ODE Solver with Empirical Model Statistics" (NeurIPS 2023)☆120Updated last year
- [ECCV 2024, Oral] FMBoost: Boosting Latent Diffusion with Flow Matching☆241Updated 8 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆358Updated this week
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆294Updated 6 months ago
- JAX implementation of MeanFlow☆319Updated 2 weeks ago
- ☆264Updated 7 months ago
- The official codebase for Reflected Flow Matching (ICML 2024)☆19Updated last year
- An official pytorch implementation of AAAI 2024 paper "Latent Space Editing in Transformer-based Flow Matching"☆42Updated last year
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆131Updated 6 months ago
- Official PyTorch implementation for the paper Minimizing Trajectory Curvature of ODE-based Generative Models, ICML 2023☆85Updated 6 months ago
- [Preprint] UCGM: Unified Continuous Generative Models☆166Updated 2 months ago
- official code for Diff-Instruct algorithm for one-step diffusion distillation☆78Updated 7 months ago
- [ICML 2025 Spotlight] Direct Discriminative Optimization: Supercharging Diffusion/Autoregressive with GAN-type Discrimination☆90Updated 2 weeks ago
- Pytorch implementation of MeanFlow on ImageNet and CIFAR10☆193Updated last week
- Official code for "Improved Techniques for Maximum Likelihood Estimation for Diffusion ODEs" (ICML 2023)☆18Updated 5 months ago
- The official implementation of "[MASK] is All You Need"☆122Updated 3 weeks ago
- The official PyTorch implementation of Fast Diffusion Model☆95Updated 2 years ago
- Official Github Repo for Neurips 2024 Paper Immiscible Diffusion: Accelerating Diffusion Training with Noise Assignment☆56Updated 2 months ago
- PyTorch re-implementation for MeanFlow☆91Updated 3 weeks ago
- ☆133Updated last year
- Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST☆71Updated 4 months ago
- [ICML'25] EQ-VAE: Equivariance Regularized Latent Space for Improved Generative Image Modeling.☆147Updated last month
- [ICML 2025] Official implementation of the paper "Compressed Image Generation with Denoising Diffusion Codebook Models"☆60Updated 2 weeks ago
- Official Implementation for Diffusion Models Without Classifier-free Guidance☆141Updated 5 months ago
- Official implementation of the paper "Unifying GANs and Score-Based Diffusion as Generative Particle Models", NeurIPS 2023☆25Updated 7 months ago