Alternatives and similar repositories for sb3_jax:

Users that are interested in sb3_jax are comparing it to the libraries listed below

• perrin-isir / xpag
  a modular reinforcement learning library with JAX agents
  ☆22Updated last year
• tseyde / decqn
  ☆35Updated 2 years ago
• ikostrikov / jaxrl2
  ☆48Updated 2 years ago
• StoneT2000 / robojax
  A high-performance reinforcement learning library in jax specialized for robotic learning
  ☆22Updated last year
• sahandrez / homomorphic_policy_gradient
  Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024
  ☆22Updated 10 months ago
• Egiob / DiversityIsAllYouNeed-SB3
  Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.
  ☆12Updated 2 years ago
• zchuning / latco
  Model-Based Reinforcement Learning via Latent-Space Collocation.
  ☆32Updated last year
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 3 years ago
• tinker495 / jax-baseline
  Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…
  ☆47Updated last week
• RyanNavillus / reward-surfaces
  ☆15Updated 9 months ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• eager-dev / eagerx_tutorials
  Tutorials on how to use EAGERx
  ☆16Updated 4 months ago
• ikostrikov / dmcgym
  ☆23Updated 2 years ago
• kevinzakka / dm_env_wrappers
  Standalone library of frequently-used wrappers for dm_env environments.
  ☆17Updated 7 months ago
• Miffyli / policy-supervectors
  Creating fixed-length vectors to describe RL/GA policies
  ☆20Updated 3 years ago
• sdpkjc / abcdrl
  Modular Single-file Reinfocement Learning Algorithms Library
  ☆37Updated last year
• ramp-kits / rl_simulator
  Model-based reinforcement learning (generative simulator models and planning agents)
  ☆15Updated 3 years ago
• philipjball / OffCon3
  📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)
  ☆24Updated 3 years ago
• kevinzakka / nanorl
  A tiny reinforcement learning codebase for continuous control, built on top of JAX.
  ☆13Updated last year
• omron-sinicx / action-constrained-RL-benchmark
  ☆23Updated 9 months ago
• pairlab / vagram
  [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.
  ☆24Updated last year
• bmazoure / ppo_jax
  Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…
  ☆53Updated 2 years ago
• mjyoo2 / offline_baselines_jax
  Offline Baselines with JAX is a set of implementations of reinforcement learning algorithms in JAX. This library is based on Stable Base…
  ☆5Updated last year
• danijar / embodied
  Fast reinforcement learning research
  ☆57Updated 2 months ago
• frt03 / inference-based-rl
  Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)
  ☆20Updated 3 years ago
• adityab / CrossQ
  Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"
  ☆64Updated 8 months ago
• qgallouedec / deep_rl
  Single-file truly minimal implementation of state-of-the-art reinforcement learning algorithms.
  ☆22Updated 2 years ago
• albertwilcox / mcac
  Author implementation of Monte Carlo Augmented Actor Critic in PyTorch
  ☆17Updated 2 years ago
• facebookresearch / bipedal-skills
  Bipedal Skills Benchmark for Reinforcement Learning
  ☆26Updated 2 years ago
• young-geng / SimpleSAC
  A simple and easy to use implementation of the soft actor-critic algorithm.
  ☆15Updated 2 years ago