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 8 months ago
• tseyde / decqn
  ☆35Updated 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
• facebookresearch / svg
  On the model-based stochastic value gradient for continuous reinforcement learning
  ☆57Updated 2 years ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆56Updated 3 years ago
• StoneT2000 / robojax
  A high-performance reinforcement learning library in jax specialized for robotic learning
  ☆22Updated 2 years ago
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated 11 months ago
• pairlab / vagram
  [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.
  ☆24Updated 2 years ago
• ikostrikov / jaxrl2
  ☆49Updated 2 years ago
• qgallouedec / deep_rl
  Single-file truly minimal implementation of state-of-the-art reinforcement learning algorithms.
  ☆21Updated 2 years ago
• ikostrikov / dmcgym
  ☆23Updated 3 years ago
• ramp-kits / rl_simulator
  Model-based reinforcement learning (generative simulator models and planning agents)
  ☆16Updated 4 years ago
• sahandrez / homomorphic_policy_gradient
  Author's PyTorch Implementation of Deep Homomorphic Policy Gradient (DHPG) - NeurIPS 2022 and JMLR 2024
  ☆23Updated last year
• Miffyli / policy-supervectors
  Creating fixed-length vectors to describe RL/GA policies
  ☆20Updated 4 years ago
• danijar / ninjax
  General Modules for JAX
  ☆69Updated last month
• Egiob / DiversityIsAllYouNeed-SB3
  Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.
  ☆12Updated 3 years ago
• MinRegret / deluca
  Performant, differentiable reinforcement learning
  ☆24Updated 2 years ago
• zafarali / emdp
  Easy MDPs and grid worlds with accessible transition dynamics to do exact calculations
  ☆49Updated 3 years ago
• 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
• RedTachyon / coltra-rl
  A modular implementation of PPO, and soon hopefully other algorithms.
  ☆26Updated last year
• 0xangelo / gym-cartpole-swingup
  A simple, continuous-control environment for OpenAI Gym
  ☆23Updated 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
• frt03 / inference-based-rl
  Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)
  ☆20Updated 4 years ago
• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆100Updated 2 years ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆29Updated 3 years ago
• nnaisense / MAGE
  Learning Action-Value Gradients in Model-based Policy Optimization
  ☆32Updated 4 years ago
• johanobandoc / revisiting_rainbow
  Revisiting Rainbow
  ☆75Updated 4 years ago