milmor / diffusion-transformer
Implementation of Diffusion Transformer Model in Pytorch
☆33Updated 4 months ago
Related projects: ⓘ
- Transformer implementation for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"☆53Updated last month
- ☆29Updated last month
- Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.☆25Updated 7 months ago
- PWM: Policy Learning with Large World Models☆32Updated last month
- Code for "DittoGym: Learning to Control Soft Shape-Shifting Robots" by Suning Huang, Boyuan Chen, Huazhe Xu, and Vincent Sitzmann.☆26Updated 4 months ago
- Source code release for "Leveraging Demonstrations with Latent Space Priors"☆39Updated last year
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆53Updated 5 months ago
- ☆24Updated last month
- ☆13Updated 7 months ago
- Slot-TTA shows that test-time adaptation using slot-centric models can improve image segmentation on out-of-distribution examples.☆23Updated last year
- ☆32Updated this week
- ☆63Updated 3 weeks ago
- ☆106Updated 10 months ago
- Neural Optimal Transport with Lagrangian Costs☆37Updated 2 months ago
- VQVAE for video prediction☆23Updated 2 years ago
- ☆39Updated 3 months ago
- Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).☆49Updated 3 weeks ago
- ☆38Updated 4 months ago
- ☆28Updated 11 months ago
- Detic + SAM for open-vocabulary object detection and segmentation.☆17Updated 3 months ago
- A minimal implementation of a denoising diffusion model in PyTorch.☆82Updated 3 months ago
- Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations☆18Updated 2 months ago
- Code for Stable Control Representations☆16Updated 3 months ago
- Implementation of Diffusion Policy, Toyota Research's supposed breakthrough in leveraging DDPMs for learning policies for real-world Robo…☆87Updated 2 months ago
- FID computation in Jax/Flax.☆23Updated 2 months ago
- ☆35Updated 2 weeks ago
- Minimal unofficial implementation of Consistency Trajectory models on a 1D toy task.☆19Updated 6 months ago
- Code for "Optimizing DDPM Sampling with Shortcut Fine-Tuning" (https://arxiv.org/abs/2301.13362), ICML 2023☆24Updated 11 months ago
- ☆19Updated 11 months ago
- Official Implementation of the ICCV 2023 paper: Perpetual Humanoid Control for Real-time Simulated Avatars in MuJoCo☆35Updated last week