Alternatives and similar repositories for JAX-RL

Users that are interested in JAX-RL are comparing it to the libraries listed below

• tinker495 / jax-baseline
  Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…
  ☆53Updated last month
• bmazoure / ppo_jax
  Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…
  ☆56Updated 2 years ago
• RPegoud / jym
  JAX implementation of RL algorithms and vectorized environments
  ☆46Updated last year
• MyNameIsArko / RL-Flax
  Various reinforcement learning algorithms written in Jax + Flax
  ☆26Updated 2 years ago
• toshikwa / rljax
  A collection of RL algorithms written in JAX.
  ☆98Updated 2 years ago
• ikostrikov / jaxrl2
  ☆47Updated 2 years ago
• Howuhh / sac-n-jax
  Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
  ☆51Updated 2 years ago
• symoon11 / dreamerv3-flax
  Flax Implementation of DreamerV3 on Crafter
  ☆16Updated 3 months ago
• yifan12wu / rl-laplacian
  Learning Laplacian Representations in Reinforcement Learning
  ☆16Updated 4 years ago
• facebookresearch / svg
  On the model-based stochastic value gradient for continuous reinforcement learning
  ☆55Updated last year
• ElisevanderPol / symmetrizer
  ☆31Updated 4 years ago
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆79Updated 3 years 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
• FLAIROx / popjym
  POPGym Library in JAX
  ☆11Updated last year
• automl / CARL
  Benchmarking RL generalization in an interpretable way.
  ☆157Updated last week
• cassidylaidlaw / effective-horizon
  Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"
  ☆48Updated last year
• moratodpg / imp_marl
  IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARL
  ☆42Updated 9 months ago
• RajGhugare19 / alm
  Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective
  ☆80Updated 2 years ago
• ahmed-touati / controllable_agent
  ☆46Updated 2 years ago
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆73Updated 10 months ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 3 years ago
• AlexGoldie / rl-learned-optimization
  Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
  ☆24Updated 2 months ago
• ethanluoyc / corax
  Corax: Core RL in JAX
  ☆38Updated last year
• Egiob / DiversityIsAllYouNeed-SB3
  Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.
  ☆12Updated 2 years ago
• facebookresearch / level-replay
  This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …
  ☆86Updated 4 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
• MichaelTMatthews / Craftax_Baselines
  ☆19Updated last month
• dojeon-ai / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆51Updated 3 weeks ago
• DramaCow / jaxued
  ☆81Updated 3 months ago
• ml-jku / OfflineRL
  ☆31Updated 2 years ago