milmor / diffusion-transformerLinks
Implementation of Diffusion Transformer Model in Pytorch
☆61Updated last month
Alternatives and similar repositories for diffusion-transformer
Users that are interested in diffusion-transformer are comparing it to the libraries listed below
Sorting:
- My take on Flow Matching☆64Updated 5 months ago
- ☆119Updated 4 months ago
- ☆23Updated last year
- PyTorch implementation of Shortcut Models [Frans, 2025] with little modification☆31Updated last month
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 9 months ago
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆126Updated last month
- Minimal unofficial implementation of Consistency Trajectory models on a 1D toy task.☆20Updated last year
- ☆17Updated 4 months ago
- ☆65Updated last year
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆114Updated 8 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆30Updated last year
- Flow-matching algorithms in JAX☆97Updated 10 months ago
- The official implementation of flow Q-learning (FQL)☆160Updated 3 months ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆51Updated 6 months ago
- ☆46Updated last year
- Implementation of Latent Diffusion Planning (Amber Xie, Oleh Rybkin, Dorsa Sadigh, Chelsea Finn)☆36Updated 2 weeks ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆66Updated 5 months ago
- ☆104Updated 2 weeks ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆221Updated 5 months ago
- A Tutorial for Diffusion Models☆53Updated last year
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆314Updated last week
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆64Updated 3 weeks ago
- Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST☆65Updated 3 months ago
- ☆9Updated 9 months ago
- [ICML 2024] Compositional Image Decomposition with Diffusion Models☆50Updated 11 months ago
- Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist☆54Updated last year
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆142Updated last year
- Annotated Flow Matching paper☆188Updated 9 months ago
- ☆33Updated 4 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆102Updated last year