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 5 months ago
• tseyde / decqn
  ☆36Updated 2 years ago
• facebookresearch / svg
  On the model-based stochastic value gradient for continuous reinforcement learning
  ☆56Updated 2 years ago
• ikostrikov / jaxrl2
  ☆47Updated 2 years ago
• sdpkjc / abcdrl
  Modular Single-file Reinfocement Learning Algorithms Library
  ☆38Updated 2 years ago
• kevinzakka / dm_env_wrappers
  Standalone library of frequently-used wrappers for dm_env environments.
  ☆17Updated last year
• Bellman-devs / bellman
  Model-based reinforcement learning in TensorFlow
  ☆56Updated 4 years ago
• StoneT2000 / robojax
  A high-performance reinforcement learning library in jax specialized for robotic learning
  ☆22Updated last year
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆56Updated 3 years ago
• pairlab / vagram
  [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.
  ☆24Updated 2 years ago
• ikostrikov / dmcgym
  ☆23Updated 2 years ago
• philipjball / OffCon3
  📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)
  ☆24Updated 4 years ago
• eager-dev / eagerx_tutorials
  Tutorials on how to use EAGERx
  ☆16Updated 9 months ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated 8 months ago
• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆100Updated last year
• roger-creus / Wave-Defense-Learning-Environment
  A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.
  ☆14Updated 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
• Michael-Beukman / RobocupGym
  Reinforcement Learning inside a 3D soccer simulation
  ☆29Updated 10 months ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆43Updated 2 years ago
• qgallouedec / deep_rl
  Single-file truly minimal implementation of state-of-the-art reinforcement learning algorithms.
  ☆21Updated 2 years ago
• cwjwudi / sin-cassie-rl-python
  ☆9Updated 2 years ago
• bmazoure / ppo_jax
  Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…
  ☆58Updated 3 years ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 3 years ago
• MinRegret / deluca
  Performant, differentiable reinforcement learning
  ☆24Updated 2 years ago
• Egiob / DiversityIsAllYouNeed-SB3
  Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.
  ☆12Updated 3 years ago
• nnaisense / MAGE
  Learning Action-Value Gradients in Model-based Policy Optimization
  ☆31Updated 3 years ago
• danijar / ninjax
  General Modules for JAX
  ☆67Updated 4 months ago
• entity-neural-network / entity-gym
  Standard interface for entity based reinforcement learning environments.
  ☆38Updated last year
• markub3327 / rl-toolkit
  RL-Toolkit: A Research Framework for Robotics
  ☆21Updated 2 weeks ago
• RedTachyon / coltra-rl
  A modular implementation of PPO, and soon hopefully other algorithms.
  ☆26Updated last year