yun-kwak / decision-transformer-jaxLinks
Decision Transformer JAX - Reproduction of 'Decision Transformer: Reinforcement Learning via Sequence Modeling' in JAX and Haiku
☆12Updated 9 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
Sorting:
- ☆16Updated 2 years ago
- Clean, extensible implementation of MACAW [ICML 2021]☆13Updated 3 years ago
- ☆45Updated 2 years ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆67Updated 2 years ago
- ☆19Updated 3 weeks ago
- POPGym Library in JAX☆11Updated last year
- Authors' PyTorch implementation of 'Recomposing the Reinforcement Learning Building-Blocks with Hypernetworks' (HypeRL)☆25Updated 3 years ago
- Learning diverse options through the Laplacian representation.☆23Updated last year
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆12Updated 10 months ago
- Mirror Descent Policy Optimization☆38Updated 4 years ago
- Official implementation of the δ-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".☆21Updated 6 months ago
- COOM: Benchmarking Continual Reinforcement Learning on Doom☆18Updated 3 months ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆17Updated last year
- Code for Model-Free Opponent Shaping (ICML 2022)☆18Updated 2 years ago
- Learning Laplacian Representations in Reinforcement Learning☆17Updated 4 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆84Updated 6 months ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆62Updated last year
- ☆17Updated 2 years ago
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆47Updated 4 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 9 months ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆57Updated last year
- ☆31Updated 2 years ago
- Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"☆30Updated 2 years ago
- Latent Dynamics Mixture, NeurIPS 2021☆18Updated 2 years ago
- Conservative Q learning in Jax☆54Updated 2 years ago
- ☆15Updated 2 years ago
- V-MPO torch version with DMLab30 and GTrXL☆13Updated 4 years ago
- ☆31Updated 4 years ago