yun-kwak / decision-transformer-jaxLinks
Decision Transformer JAX - Reproduction of 'Decision Transformer: Reinforcement Learning via Sequence Modeling' in JAX and Haiku
☆12Updated last year
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:
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆70Updated 3 years ago
- Clean, extensible implementation of MACAW [ICML 2021]☆12Updated 4 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆112Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆22Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 3 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆103Updated 3 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆62Updated 2 weeks ago
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Updated last year
- Challenging Memory-based Deep Reinforcement Learning Agents☆109Updated last year
- Learning diverse options through the Laplacian representation.☆23Updated 2 years ago
- [ICLR 2023 Oral] The official implementation of SQL and EQL in "Offline RL with No OOD Actions: In-Sample Learning via Implicit Value Reg…☆47Updated 2 years ago
- Learning Laplacian Representations in Reinforcement Learning☆18Updated 5 years ago
- ☆18Updated 3 years ago
- V-MPO torch version with DMLab30 and GTrXL☆13Updated 4 years ago
- ☆17Updated 3 years ago
- COOM: Benchmarking Continual Reinforcement Learning on Doom☆21Updated 2 weeks ago
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆48Updated 5 years ago
- Official implementation of Harnessing Mixed Offline Reinforcement Learning Datasets via Trajectory Reweighting☆17Updated last year
- Extreme Q-Learning: Max Entropy RL without Entropy☆87Updated 2 years ago
- Conservative Q learning in Jax☆57Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆124Updated 3 years ago
- Official codebase for GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning.☆29Updated last year
- AGAC: Adversarially Guided Actor-Critic☆47Updated 4 years ago
- ☆31Updated 3 years ago
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Updated last month
- on-policy optimization baselines for deep reinforcement learning☆32Updated 5 years ago
- ☆20Updated 8 months ago
- Official codebase for Generating Diverse Cooperative Agents by Learning Incompatible Policies (notable-top-25% @ ICLR 2023)☆19Updated last year
- Synthetic Experience Replay☆107Updated last year
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆13Updated last year