milmor / diffusion-transformerLinks
Implementation of Diffusion Transformer Model in Pytorch
☆60Updated 3 weeks ago
Alternatives and similar repositories for diffusion-transformer
Users that are interested in diffusion-transformer are comparing it to the libraries listed below
Sorting:
- PyTorch implementation of Shortcut Models [Frans, 2025] with little modification☆27Updated 3 weeks ago
- ☆22Updated last year
- My take on Flow Matching☆57Updated 4 months ago
- ☆118Updated 3 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆115Updated 7 months ago
- ☆65Updated 11 months ago
- A minimal implementation of a denoising diffusion model in PyTorch.☆122Updated 6 months ago
- Minimal unofficial implementation of Consistency Trajectory models on a 1D toy task.☆20Updated last year
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 8 months ago
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆138Updated last year
- ☆14Updated 3 months ago
- This repo implements VQVAE on mnist and as well as colored version of mnist images. It also implements simple LSTM for generating sample …☆56Updated last year
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆66Updated 4 months ago
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 6 months ago
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆123Updated last month
- The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)☆76Updated 5 months ago
- Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST☆59Updated 2 months ago
- A Tutorial for Diffusion Models☆52Updated last year
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆294Updated last week
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆63Updated this week
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆29Updated last year
- The aim of this repository is to test and implement Flow-Matching-based models☆92Updated 4 months ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆16Updated 10 months ago
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆62Updated last year
- ☆188Updated 5 months ago
- ☆46Updated last year
- Diffusing States and Matching Scores: A New Framework for Imitation Learning☆19Updated 6 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆92Updated 6 months ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆216Updated 4 months ago
- Official PyTorch Implementation of "Diffusion Autoencoders are Scalable Image Tokenizers"☆121Updated 4 months ago