milmor / diffusion-transformerLinks
Implementation of Diffusion Transformer Model in Pytorch
☆62Updated 2 months ago
Alternatives and similar repositories for diffusion-transformer
Users that are interested in diffusion-transformer are comparing it to the libraries listed below
Sorting:
- A minimal implementation of a denoising diffusion model in PyTorch.☆124Updated 8 months ago
- PyTorch implementation of Shortcut Models [Frans, 2025] with little modification☆41Updated last week
- ☆120Updated 4 months ago
- Code for "DittoGym: Learning to Control Soft Shape-Shifting Robots" by Suning Huang, Boyuan Chen, Huazhe Xu, and Vincent Sitzmann.☆28Updated 2 months ago
- My take on Flow Matching☆66Updated 6 months ago
- Implementation of Diffusion Policy, Toyota Research's supposed breakthrough in leveraging DDPMs for learning policies for real-world Robo…☆113Updated last year
- Hierarchical State Space Models☆47Updated last year
- ☆24Updated last year
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆128Updated 2 months ago
- ☆67Updated last year
- Slot-TTA shows that test-time adaptation using slot-centric models can improve image segmentation on out-of-distribution examples.☆26Updated 2 years ago
- VQVAE for video prediction☆27Updated 3 years ago
- Implementation of a transformer for reinforcement learning using `x-transformers`☆61Updated last month
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆18Updated last year
- Implementation of Latent Diffusion Planning (Amber Xie, Oleh Rybkin, Dorsa Sadigh, Chelsea Finn)☆38Updated 2 weeks ago
- Code for "Optimizing DDPM Sampling with Shortcut Fine-Tuning" (https://arxiv.org/abs/2301.13362), ICML 2023☆30Updated last year
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆97Updated last week
- ☆18Updated 5 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 9 months ago
- Implementation of a framework for Genie2 in Pytorch☆149Updated 6 months ago
- Diffusing States and Matching Scores: A New Framework for Imitation Learning☆20Updated 8 months ago
- [CVPR 2025] Parallel Sequence Modeling via Generalized Spatial Propagation Network☆100Updated 2 weeks ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆67Updated 5 months ago
- Code for the paper "Learning a Diffusion Model Policy from Rewards via Q-Score Matching"☆23Updated 3 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆65Updated last month
- ☆41Updated last year
- A simple implementation of classifier-free guidance DDIM on MNIST☆33Updated 2 years ago
- ☆76Updated last month
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆66Updated last year
- [NeurIPS 2024] GenRL: Multimodal-foundation world models enable grounding language and video prompts into embodied domains, by turning th…☆78Updated 3 months ago