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:
- ☆129Updated this week
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆871Oct 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 3 weeks ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆21Jul 14, 2024Updated last year
- DreamSmooth: Improving Model-Based RL with Reward Smoothing (ICLR 2024)☆12May 6, 2024Updated last year
- Implementation of Dreamer v3 in pytorch.☆820Mar 8, 2026Updated 2 weeks ago
- Evaluation of TD-MPC2.☆21Jan 21, 2024Updated 2 years ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆30Oct 12, 2023Updated 2 years ago
- This repo support auto line plot for multi-seed event file from TensorBoard☆12Jun 23, 2022Updated 3 years ago
- [ICLR 2025] Learning Transformer-based World Models with Contrastive Predictive Coding (TWISTER)☆50Mar 9, 2025Updated last year
- Reproduction of Dreamerv1 and v2 in pytorch for deepmind control suite☆49Dec 27, 2022Updated 3 years ago
- [ECCV2022] [T-PAMI] StARformer: Transformer with State-Action-Reward Representations.☆96May 21, 2023Updated 2 years ago
- Pytorch version of Dreamer, which follows the original TF v2 codes.☆140Feb 7, 2022Updated 4 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
- IPHYRE: Interactive Physical Reasoning, ICLR 2024☆19Oct 18, 2024Updated last year
- Deep Learning Project☆23Jan 18, 2020Updated 6 years ago
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆163Dec 16, 2025Updated 3 months ago
- Evaluating long-term memory of reinforcement learning algorithms☆165Jun 23, 2023Updated 2 years ago
- [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
- PWM: Policy Learning with Large World Models☆65Aug 4, 2025Updated 7 months ago
- ☆17Nov 18, 2024Updated last year
- Implementation of Tactical Optimistic and Pessimistic value estimation☆25Jul 18, 2023Updated 2 years ago
- Pytorch implementation of DreamerV2: Mastering Atari with Discrete World Models, based on the original implementation☆22Jul 25, 2022Updated 3 years ago
- Code for RRL (https://sites.google.com/view/abstractions4rl)☆27Jan 21, 2022Updated 4 years ago
- Learning bisimulation metrics for control, particularly suited to sparse reward settings☆10Feb 28, 2023Updated 3 years ago
- Official PyTorch implementation of "DC-MPC: Discrete Codebook Model Predictive Control"☆31Mar 15, 2025Updated last year
- Code for "Learning to Model the World with Language." ICML 2024 Oral.☆414Jan 7, 2026Updated 2 months ago
- Offline RLHF codebase implementation for "Uni-RLHF: Universal Platform and Benchmark Suite for Reinforcement Learning with Diverse Human …☆41Mar 26, 2024Updated last year
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 2 years ago
- ☆130Feb 22, 2025Updated last year
- OpenAI Gym wrapper for the DeepMind Control Suite☆229May 19, 2024Updated last year
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆119Sep 22, 2024Updated last year
- Meta-RL Model-Based Algorithm☆44Apr 30, 2025Updated 10 months ago
- Mastering Atari with Discrete World Models☆1,014Jan 21, 2023Updated 3 years ago
- ☆17Dec 21, 2020Updated 5 years ago
- ☆36May 24, 2023Updated 2 years ago