Alternatives and similar repositories for decision-transformer-jax

Users that are interested in decision-transformer-jax are comparing it to the libraries listed below

• frt03 / generalized_dt
  Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)
  ☆67Updated 2 years ago
• frt03 / jax_dt
  Minimal Decision Transformer Implementation written in Jax (Flax).
  ☆17Updated 2 years ago
• RajGhugare19 / stitching-is-combinatorial-generalisation
  [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.
  ☆23Updated last year
• eric-mitchell / macaw-min
  Clean, extensible implementation of MACAW [ICML 2021]
  ☆12Updated 3 years ago
• evgenii-nikishin / rl_with_resets
  JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"
  ☆100Updated 3 years ago
• jsikyoon / V-MPO_torch
  V-MPO torch version with DMLab30 and GTrXL
  ☆13Updated 4 years ago
• ryanxhr / IVR
  [ICLR 2023 Oral] The official implementation of SQL and EQL in "Offline RL with No OOD Actions: In-Sample Learning via Implicit Value Reg…
  ☆44Updated last year
• ahmed-touati / controllable_agent
  ☆46Updated 2 years ago
• denisyarats / exorl
  ExORL: Exploratory Data for Offline Reinforcement Learning
  ☆115Updated 3 years ago
• MarcoMeter / endless-memory-gym
  Challenging Memory-based Deep Reinforcement Learning Agents
  ☆102Updated 8 months ago
• dojeon-ai / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆57Updated last month
• twni2016 / Memory-RL
  When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)
  ☆63Updated last year
• keynans / HypeRL
  Authors' PyTorch implementation of 'Recomposing the Reinforcement Learning Building-Blocks with Hypernetworks' (HypeRL)
  ☆24Updated 4 years ago
• YYCAAA / V-MPO_Lunarlander
  Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238
  ☆48Updated 4 years ago
• ml-jku / OfflineRL
  ☆31Updated 2 years ago
• yifan12wu / rl-laplacian
  Learning Laplacian Representations in Reinforcement Learning
  ☆17Updated 4 years ago
• lanyavik / BAIL
  ☆17Updated 3 years ago
• Jaewoopudding / GTA
  Official codebase for GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning.
  ☆22Updated 8 months ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆107Updated last year
• micahcarroll / uniMASK
  Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"
  ☆56Updated last year
• uncharted-technologies / risk-and-uncertainty
  Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"
  ☆30Updated 2 years ago
• young-geng / JaxCQL
  Conservative Q learning in Jax
  ☆54Updated 2 years ago
• RajGhugare19 / alm
  Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective
  ☆81Updated 2 years ago
• seohongpark / METRA
  METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)
  ☆71Updated last year
• Div99 / XQL
  Extreme Q-Learning: Max Entropy RL without Entropy
  ☆87Updated 2 years ago
• symoon11 / dreamerv3-flax
  Flax Implementation of DreamerV3 on Crafter
  ☆16Updated 4 months ago
• sail-sg / rosmo
  Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023
  ☆29Updated 2 years ago
• fdeng18 / dreamer-pro
  ☆36Updated 2 years ago
• scottemmons / rvs
  Reinforcement Learning via Supervised Learning
  ☆71Updated 3 years ago
• dibyaghosh / jaxrl_m
  Skeleton for scalable and flexible Jax RL implementations
  ☆83Updated 2 years ago