yun-kwak / decision-transformer-jax
Decision Transformer JAX - Reproduction of 'Decision Transformer: Reinforcement Learning via Sequence Modeling' in JAX and Haiku
☆12Updated 5 months ago
Alternatives and similar repositories for decision-transformer-jax:
Users that are interested in decision-transformer-jax are comparing it to the libraries listed below
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated 9 months ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Official codebase for GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning.☆17Updated 2 months ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆66Updated 2 years ago
- Clean, extensible implementation of MACAW [ICML 2021]☆13Updated 3 years ago
- ☆27Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 3 years ago
- ☆41Updated last year
- CaDM: Context-aware Dynamics Model for Generalization in Model-based Reinforcement Learning☆63Updated 4 years ago
- Authors' PyTorch implementation of 'Recomposing the Reinforcement Learning Building-Blocks with Hypernetworks' (HypeRL)☆25Updated 3 years ago
- Official implementation of Harnessing Mixed Offline Reinforcement Learning Datasets via Trajectory Reweighting☆16Updated 11 months ago
- ☆29Updated 2 years ago
- ☆14Updated last year
- V-MPO torch version with DMLab30 and GTrXL☆12Updated 3 years ago
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆53Updated 6 months ago
- Code for FOCAL Paper Published at ICLR 2021☆51Updated last year
- [ICLR 2023 Oral] The official implementation of SQL and EQL in "Offline RL with No OOD Actions: In-Sample Learning via Implicit Value Reg…☆45Updated last year
- Learning Laplacian Representations in Reinforcement Learning☆17Updated 4 years ago
- ☆47Updated last year
- Learning diverse options through the Laplacian representation.☆23Updated last year
- ☆17Updated 2 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated last month
- ☆18Updated last year
- Synthetic Experience Replay☆84Updated 8 months ago
- EARL: Environment for Autonomous Reinforcement Learning☆34Updated 2 years ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆54Updated 10 months ago
- Code for NeurIPS 2021 paper "Curriculum Offline Imitation Learning"☆17Updated 2 years ago
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆44Updated 4 years ago
- TensorFlow implementation for our paper "Learning Long-Term Reward Redistribution via Randomized Return Decomposition"☆19Updated 2 years ago
- Conservative Q learning in Jax☆52Updated last year