jrobine / twm
Transformer-based World Models
☆75Updated last year
Alternatives and similar repositories for twm:
Users that are interested in twm are comparing it to the libraries listed below
- ☆69Updated 7 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated last month
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆98Updated 7 months ago
- Official code repository for Prompt-DT.☆101Updated 2 years ago
- Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.☆101Updated last year
- Source files to replicate experiments in my ICLR 2022 paper.☆67Updated 6 months ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆77Updated 8 months ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆23Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆63Updated 7 months ago
- official implementation for our paper Cal-QL: Calibrated Offline RL Pre-Training for Efficient Online Fine-Tuning☆81Updated 5 months ago
- Implementation of Jump-Start Reinforcement Learning (JSRL) with Stable Baselines3☆28Updated last year
- ☆69Updated 3 months ago
- Synthetic Experience Replay☆82Updated 7 months ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆59Updated last year
- OGBench: Benchmarking Offline Goal-Conditioned RL☆99Updated 2 months ago
- This repo relates to the survey paper <Goal-Conditioned Reinforcement Learning: Problems and Solutions>. We collects widely used benchmar…☆117Updated last year
- Extreme Q-Learning: Max Entropy RL without Entropy☆82Updated last year
- Pytorch version of Dreamer, which follows the original TF v2 codes.☆121Updated 2 years ago
- DrM, a visual RL algorithm, minimizes the dormant ratio to guide exploration-exploitation trade-offs, achieving significant improvements …☆69Updated 7 months ago
- ☆69Updated 2 years ago
- ☆47Updated last year
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆85Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆76Updated 9 months ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated last year
- Skeleton for scalable and flexible Jax RL implementations☆67Updated last year
- ☆20Updated 8 months ago
- Masked World Models for Visual Control☆119Updated last year
- Author's PyTorch implementation of TD7 for online and offline RL☆122Updated last year
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆34Updated 10 months ago
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆44Updated 10 months ago