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
  ☆27Updated 6 months ago
• sdpkjc / abcdrl
  Modular Single-file Reinfocement Learning Algorithms Library
  ☆38Updated 2 years ago
• tseyde / decqn
  ☆35Updated 2 years ago
• kevinzakka / dm_env_wrappers
  Standalone library of frequently-used wrappers for dm_env environments.
  ☆17Updated last year
• ikostrikov / jaxrl2
  ☆46Updated 2 years ago
• Bellman-devs / bellman
  Model-based reinforcement learning in TensorFlow
  ☆56Updated 4 years ago
• facebookresearch / svg
  On the model-based stochastic value gradient for continuous reinforcement learning
  ☆56Updated 2 years ago
• eager-dev / eagerx_tutorials
  Tutorials on how to use EAGERx
  ☆16Updated last month
• StoneT2000 / robojax
  A high-performance reinforcement learning library in jax specialized for robotic learning
  ☆22Updated 2 years ago
• ikostrikov / dmcgym
  ☆23Updated 3 years ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated 9 months ago
• sahandrez / homomorphic_policy_gradient
  Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024
  ☆23Updated last year
• RyanNavillus / reward-surfaces
  ☆18Updated last year
• pairlab / vagram
  [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.
  ☆24Updated 2 years ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆56Updated 3 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
• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆100Updated last year
• MinRegret / deluca
  Performant, differentiable reinforcement learning
  ☆24Updated 2 years ago
• ingambe / RayEnvWrapper
  OpenAi's gym environment wrapper to vectorize them with Ray
  ☆23Updated 2 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆43Updated 3 years ago
• Yuxing-Wang-THU / ModularEvoGym
  A modified benchmark for designing and controlling 2D Voxel-based Soft Robots
  ☆35Updated last year
• nnaisense / MAGE
  Learning Action-Value Gradients in Model-based Policy Optimization
  ☆31Updated 4 years ago
• danijar / ninjax
  General Modules for JAX
  ☆67Updated 2 weeks ago
• RedTachyon / coltra-rl
  A modular implementation of PPO, and soon hopefully other algorithms.
  ☆26Updated last year
• clvrai / coordination
  Learning to Coordinate Manipulation Skills via Skill Behavior Diversification (ICLR 2020)
  ☆48Updated 3 years ago
• qgallouedec / deep_rl
  Single-file truly minimal implementation of state-of-the-art reinforcement learning algorithms.
  ☆21Updated 2 years ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 3 years ago
• hari-sikchi / LOOP
  Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]
  ☆40Updated 3 years ago
• famura / SimuRLacra
  reinforcement learning from randomized simulations
  ☆69Updated 5 months ago