Transformer-based World Models
☆90Apr 4, 2023Updated 3 years ago
Alternatives and similar repositories for twm
Users that are interested in twm are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆903Oct 14, 2024Updated 2 years ago
- ☆141Mar 18, 2026Updated 7 months ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆33Oct 12, 2023Updated 3 years ago
- Masked World Models for Visual Control☆139Jun 11, 2023Updated 3 years ago
- Improving Token-Based World Models with Parallel Observation Prediction (ICML 2024)☆15Feb 23, 2026Updated 8 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Reproduction of Dreamerv1 and v2 in pytorch for deepmind control suite☆50Dec 27, 2022Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 11 months ago
- Implementation of Dreamer v3 in pytorch.☆888Mar 8, 2026Updated 7 months ago
- [ICLR 2025] Learning Transformer-based World Models with Contrastive Predictive Coding (TWISTER)☆60Mar 9, 2025Updated last year
- Evaluation of TD-MPC2.☆21Jan 21, 2024Updated 2 years ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆23Jul 14, 2024Updated 2 years ago
- DreamSmooth: Improving Model-Based RL with Reward Smoothing (ICLR 2024)☆13May 6, 2024Updated 2 years ago
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 3 years ago
- Pytorch implementation of DreamerV2: Mastering Atari with Discrete World Models, based on the original implementation☆22Jul 25, 2022Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Pytorch Implementation of Learning Latent Dynamic Robust Representations for World Models☆27May 11, 2024Updated 2 years ago
- This repo support auto line plot for multi-seed event file from TensorBoard☆12Jun 23, 2022Updated 4 years ago
- [ECCV2022] [T-PAMI] StARformer: Transformer with State-Action-Reward Representations.☆97May 21, 2023Updated 3 years ago
- PyTorch World Model implementation with PPO.☆13Jan 6, 2020Updated 6 years ago
- ☆18Aug 15, 2025Updated last year
- 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.☆276Jul 29, 2023Updated 3 years ago
- Pytorch version of Dreamer, which follows the original TF v2 codes.☆143Feb 7, 2022Updated 4 years ago
- Efficient World Models with Context-Aware Tokenization. ICML 2024☆133Sep 22, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆26Jan 26, 2024Updated 2 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆186Jun 23, 2023Updated 3 years ago
- PWM: Policy Learning with Large World Models☆71Aug 4, 2025Updated last year
- OpenAI Gym wrapper for the DeepMind Control Suite☆228May 19, 2024Updated 2 years ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆42Feb 27, 2024Updated 2 years ago
- Deep Learning Project☆23Jan 18, 2020Updated 6 years ago
- [ICLR 2025] Bootstrapped Model Predictive Control☆41Jul 20, 2026Updated 3 months ago
- ☆25Sep 23, 2024Updated 2 years ago
- Code for RRL (https://sites.google.com/view/abstractions4rl)☆27Jan 21, 2022Updated 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- IPHYRE: Interactive Physical Reasoning, ICLR 2024☆19Oct 18, 2024Updated 2 years ago
- Code release for "HarmonyDream: Task Harmonization Inside World Models" (ICML 2024), https://arxiv.org/abs/2310.00344☆57Jun 27, 2024Updated 2 years ago
- MTM Masked Trajectory Models for Prediction, Representation, and Control.☆167Dec 16, 2025Updated 10 months ago
- Code for "Learning to Model the World with Language." ICML 2024 Oral.☆420Jan 7, 2026Updated 9 months ago
- [ICML 2024, Spotlight] EfficientZero V2: Mastering Discrete and Continuous Control with Limited Data☆125Aug 9, 2024Updated 2 years ago
- A PyTorch port of DeepMind's Disco103 — the meta-learned reinforcement learning update rule from Discovering State-of-the-art Reinforceme…☆15Sep 3, 2026Updated last month
- Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"☆993Oct 20, 2026Updated last week