Transformer-based World Models
☆89Apr 4, 2023Updated 2 years ago
Alternatives and similar repositories for twm
Users that are interested in twm are comparing it to the libraries listed below
Sorting:
- ☆126Nov 25, 2025Updated 3 months ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆869Oct 14, 2024Updated last year
- Masked World Models for Visual Control☆135Jun 11, 2023Updated 2 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 3 months ago
- Improving Token-Based World Models with Parallel Observation Prediction (ICML 2024)☆14Feb 23, 2026Updated last week
- Implementation of Dreamer v3 in pytorch.☆797Feb 21, 2026Updated last week
- Evaluation of TD-MPC2.☆21Jan 21, 2024Updated 2 years ago
- Reproduction of Dreamerv1 and v2 in pytorch for deepmind control suite☆48Dec 27, 2022Updated 3 years ago
- DreamSmooth: Improving Model-Based RL with Reward Smoothing (ICLR 2024)☆12May 6, 2024Updated last year
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆21Jul 14, 2024Updated last year
- [ECCV2022] [T-PAMI] StARformer: Transformer with State-Action-Reward Representations.☆96May 21, 2023Updated 2 years ago
- Predictable MDP Abstraction for Unsupervised Model-Based RL (ICML 2023)☆32Feb 6, 2023Updated 3 years ago
- Pytorch implementation of Dreamer-v2: Visual Model Based RL Algorithm.☆274Jul 29, 2023Updated 2 years ago
- This repo support auto line plot for multi-seed event file from TensorBoard☆12Jun 23, 2022Updated 3 years ago
- Offline RLHF codebase implementation for "Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human …☆41Mar 26, 2024Updated last year
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆41Feb 27, 2024Updated 2 years ago
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆162Dec 16, 2025Updated 2 months ago
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 2 years ago
- Deep Learning Project☆23Jan 18, 2020Updated 6 years ago
- Code for RRL (https://sites.google.com/view/abstractions4rl)☆27Jan 21, 2022Updated 4 years ago
- PyTorch World Model implementation with PPO.☆13Jan 6, 2020Updated 6 years ago
- ☆15Aug 9, 2021Updated 4 years ago
- Pytorch version of Dreamer, which follows the original TF v2 codes.☆140Feb 7, 2022Updated 4 years ago
- IPHYRE: Interactive Physical Reasoning, ICLR 2024☆19Oct 18, 2024Updated last year
- Evaluating long-term memory of reinforcement learning algorithms☆164Jun 23, 2023Updated 2 years ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆29Oct 12, 2023Updated 2 years ago
- Reproducing results of Domain Randomization for Transferring Deep Neural Networks from Simulation to the Real World https://arxiv.org/abs…☆27Dec 27, 2022Updated 3 years ago
- [ICLR 2025] Learning Transformer-based World Models with Contrastive Predictive Coding (TWISTER)☆49Mar 9, 2025Updated 11 months ago
- Official PyTorch implementation of "DC-MPC: Discrete Codebook Model Predictive Control"☆31Mar 15, 2025Updated 11 months ago
- ☆17Dec 21, 2020Updated 5 years ago
- Code for "Learning to Model the World with Language." ICML 2024 Oral.☆413Jan 7, 2026Updated last month
- A Simplified Pytorch Version of the Dreamer Algorithm☆149Jul 24, 2023Updated 2 years ago
- DrQ-v2: Improved Data-Augmented Reinforcement Learning☆431May 31, 2022Updated 3 years ago
- OpenAI Gym wrapper for the DeepMind Control Suite☆227May 19, 2024Updated last year
- Pytorch implementations of RL algorithms, focusing on model-based, lifelong, reset-free, and offline algorithms. Official codebase for Re…☆110Jan 23, 2022Updated 4 years ago
- Mastering Atari with Discrete World Models☆1,009Jan 21, 2023Updated 3 years ago
- Pytorch implementation of DreamerV2: Mastering Atari with Discrete World Models, based on the original implementation☆22Jul 25, 2022Updated 3 years ago
- Task planning over 3D scene graphs☆19Jul 8, 2022Updated 3 years ago
- MoDem Accelerating Visual Model-Based Reinforcement Learning with Demonstrations☆86Dec 12, 2022Updated 3 years ago