milmor / diffusion-transformer
Implementation of Diffusion Transformer Model in Pytorch
☆53Updated 3 months ago
Alternatives and similar repositories for diffusion-transformer:
Users that are interested in diffusion-transformer are comparing it to the libraries listed below
- ☆58Updated 9 months ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆57Updated last month
- ☆116Updated last month
- ☆20Updated last year
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆26Updated 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☆97Updated this week
- VQVAE for video prediction☆27Updated 2 years ago
- Official repository for "iVideoGPT: Interactive VideoGPTs are Scalable World Models" (NeurIPS 2024), https://arxiv.org/abs/2405.15223☆121Updated 3 weeks ago
- ☆75Updated 7 months ago
- The official implementation of flow Q-learning (FQL)☆123Updated 2 weeks ago
- Explorations into improving ViTArc with Slot Attention☆39Updated 5 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated 6 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆59Updated 2 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 …☆52Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 3 months ago
- Implementation of a multimodal diffusion transformer in Pytorch☆101Updated 9 months ago
- Implementation of the proposed Spline-Based Transformer from Disney Research☆87Updated 4 months ago
- Hierarchical State Space Models☆44Updated 11 months ago
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆80Updated this week
- ☆413Updated 3 months ago
- A Tutorial for Diffusion Models☆45Updated last year
- ElasticTok: Adaptive Tokenization for Image and Video☆64Updated 4 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆112Updated last month
- Slot-TTA shows that test-time adaptation using slot-centric models can improve image segmentation on out-of-distribution examples.☆26Updated last year
- ☆41Updated 10 months ago
- Implementation of Diffusion Policy, Toyota Research's supposed breakthrough in leveraging DDPMs for learning policies for real-world Robo…☆105Updated 8 months ago
- A Video Tokenizer Evaluation Dataset☆109Updated 2 months ago
- Pytorch implementation of "Genie: Generative Interactive Environments", Bruce et al. (2024).☆143Updated 7 months ago
- Code release for "Pre-training Contextualized World Models with In-the-wild Videos for Reinforcement Learning" (NeurIPS 2023), https://ar…☆58Updated 6 months ago
- An ML research template with good documentation by Boyuan Chen, an MIT PhD student☆64Updated 3 weeks ago