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
  ☆26Updated 4 months ago
• tseyde / decqn
  ☆36Updated 2 years ago
• StoneT2000 / robojax
  A high-performance reinforcement learning library in jax specialized for robotic learning
  ☆22Updated last year
• kevinzakka / dm_env_wrappers
  Standalone library of frequently-used wrappers for dm_env environments.
  ☆17Updated last year
• ikostrikov / dmcgym
  ☆23Updated 2 years ago
• sdpkjc / abcdrl
  Modular Single-file Reinfocement Learning Algorithms Library
  ☆38Updated 2 years ago
• facebookresearch / svg
  On the model-based stochastic value gradient for continuous reinforcement learning
  ☆55Updated 2 years ago
• sahandrez / homomorphic_policy_gradient
  Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024
  ☆23Updated last year
• Bellman-devs / bellman
  Model-based reinforcement learning in TensorFlow
  ☆56Updated 3 years ago
• qgallouedec / deep_rl
  Single-file truly minimal implementation of state-of-the-art reinforcement learning algorithms.
  ☆21Updated 2 years ago
• eager-dev / eagerx_tutorials
  Tutorials on how to use EAGERx
  ☆16Updated 9 months ago
• ikostrikov / jaxrl2
  ☆47Updated 2 years ago
• Egiob / DiversityIsAllYouNeed-SB3
  Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.
  ☆12Updated 3 years ago
• philipjball / OffCon3
  📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)
  ☆24Updated 4 years ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated 7 months ago
• pairlab / vagram
  [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.
  ☆24Updated 2 years ago
• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆100Updated last year
• ingambe / RayEnvWrapper
  OpenAi's gym environment wrapper to vectorize them with Ray
  ☆23Updated 2 years ago
• MinRegret / deluca
  Performant, differentiable reinforcement learning
  ☆24Updated 2 years ago
• bmazoure / ppo_jax
  Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…
  ☆57Updated 2 years ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆56Updated 2 years ago
• RedTachyon / coltra-rl
  A modular implementation of PPO, and soon hopefully other algorithms.
  ☆26Updated last year
• famura / SimuRLacra
  reinforcement learning from randomized simulations
  ☆69Updated 3 months ago
• Yuxing-Wang-THU / ModularEvoGym
  A modified benchmark for designing and controlling 2D Voxel-based Soft Robots
  ☆35Updated last year
• ademiadeniji / irm
  Intrinsic Reward Matching (IRM) implementation (from Adeniji and Xie et al 2022)
  ☆43Updated last year
• frt03 / inference-based-rl
  Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)
  ☆20Updated 3 years ago
• kevinzakka / nanorl
  A tiny reinforcement learning codebase for continuous control, built on top of JAX.
  ☆13Updated 2 years ago
• hari-sikchi / LOOP
  Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]
  ☆40Updated 2 years ago
• RyanNavillus / reward-surfaces
  ☆17Updated last year
• ramp-kits / rl_simulator
  Model-based reinforcement learning (generative simulator models and planning agents)
  ☆16Updated 3 years ago