Howuhh / faster-trajectory-transformerLinks
Implementation of Trajectory Transformer with attention caching and batched beam search
☆115Updated 2 years ago
Alternatives and similar repositories for faster-trajectory-transformer
Users that are interested in faster-trajectory-transformer are comparing it to the libraries listed below
Sorting:
- ExORL: Exploratory Data for Offline Reinforcement Learning☆115Updated 3 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆100Updated 3 years ago
- pytorch-implementation of Dreamer (Model-based Image RL Algorithm)☆166Updated 7 months ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆227Updated 2 years ago
- Learning Invariant Representations for Reinforcement Learning without Reconstruction☆150Updated 4 years ago
- Code for MOPO: Model-based Offline Policy Optimization☆188Updated 3 years ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆67Updated 3 years ago
- Extreme Q-Learning: Max Entropy RL without Entropy☆87Updated 2 years ago
- Pytorch version of Dreamer, which follows the original TF v2 codes.☆131Updated 3 years ago
- ☆201Updated 2 years ago
- ☆282Updated 3 years ago
- Conservative Q Learning on top of SAC☆132Updated 2 years ago
- Clean baseline implementation of PPO using an episodic TransformerXL memory☆187Updated last year
- Challenging Memory-based Deep Reinforcement Learning Agents☆103Updated 10 months ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated 2 years ago
- Datasets for data-driven deep reinforcement learning with Atari (wrapper for datasets released by Google)☆122Updated last year
- Code for "Data-Efficient Reinforcement Learning with Self-Predictive Representations"☆162Updated 3 years ago
- Benchmarking RL generalization in an interpretable way.☆161Updated 2 months ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆65Updated last year
- Model-Based Offline Reinforcement Learning☆51Updated 4 years ago
- Repo for Implicit Diffusion Q-Learning☆114Updated last year
- A pytorch reprelication of the model-based reinforcement learning algorithm MBPO☆180Updated 3 years ago
- Deep Hierarchical Planning from Pixels☆107Updated 2 years ago
- Reinforcement Learning via Supervised Learning☆71Updated 3 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆88Updated 9 months ago
- Author's PyTorch implementation of TD7 for online and offline RL☆148Updated 2 years ago
- Synthetic Experience Replay☆102Updated last year
- Official PyTorch implementation of "Uncertainty-Based Offline Reinforcement Learning with Diversified Q-Ensemble" (NeurIPS'21)☆76Updated 3 years ago
- ☆48Updated 2 years ago
- Partially Observable Process Gym☆198Updated 3 months ago