Alternatives and similar repositories for decision-transformer-jax:

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

• ElisevanderPol / symmetrizer
  ☆29Updated 4 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 10 months ago
• frt03 / generalized_dt
  Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)
  ☆67Updated 2 years ago
• frt03 / mxt_bench
  A System for Morphology-Task Generalization via Unified Representation and Behavior Distillation (ICLR2023)
  ☆13Updated 2 years ago
• mklissa / dceo
  Learning diverse options through the Laplacian representation.
  ☆23Updated last year
• ml-jku / OfflineRL
  ☆29Updated 2 years ago
• wangzizhao / CausalDynamicsLearning
  ☆29Updated 2 years ago
• ahmed-touati / controllable_agent
  ☆43Updated last year
• eric-mitchell / macaw-min
  Clean, extensible implementation of MACAW [ICML 2021]
  ☆13Updated 3 years ago
• amazon-science / replay-based-recurrent-rl
  Code for "Task-Agnostic Continual RL: In Praise of a Simple Baseline"
  ☆33Updated last year
• MichaelTMatthews / Craftax_Baselines
  ☆18Updated last month
• younggyoseo / CaDM
  CaDM: Context-aware Dynamics Model for Generalization in Model-based Reinforcement Learning
  ☆63Updated 4 years ago
• ucl-dark / skillhack
  SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning
  ☆14Updated 2 years ago
• young-geng / JaxCQL
  Conservative Q learning in Jax
  ☆52Updated 2 years ago
• FLAIROx / popjym
  POPGym Library in JAX
  ☆11Updated 10 months ago
• yardenas / la-mbda
  LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization
  ☆34Updated 2 years ago
• TTomilin / COOM
  COOM: Benchmarking Continual Reinforcement Learning on Doom
  ☆15Updated 2 weeks ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 3 years ago
• seohongpark / HIQL
  HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)
  ☆76Updated 3 months ago
• oist-cnru / Variational-Recurrent-Models
  Codes for the study "Variational Recurrent Models for Solving Partially Observable Control Tasks", published as a conference paper at ICL…
  ☆53Updated 4 years ago
• samlobel / CFN
  Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023
  ☆19Updated last year
• lmzintgraf / hyperx
  ☆16Updated 2 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 2 years ago
• namsan96 / SiMPL
  ☆48Updated last year
• yifan12wu / rl-laplacian
  Learning Laplacian Representations in Reinforcement Learning
  ☆17Updated 4 years ago
• YYCAAA / V-MPO_Lunarlander
  Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238
  ☆44Updated 4 years ago
• hwang-ua / inac_pytorch
  ☆18Updated last year
• google-deepmind / active_ops
  ☆32Updated 7 months ago
• facebookresearch / gen_dgrl
  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024
  ☆28Updated 7 months ago