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 5 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
• facebookresearch / svg
  On the model-based stochastic value gradient for continuous reinforcement learning
  ☆55Updated 2 years ago
• sdpkjc / abcdrl
  Modular Single-file Reinfocement Learning Algorithms Library
  ☆38Updated 2 years ago
• RedTachyon / coltra-rl
  A modular implementation of PPO, and soon hopefully other algorithms.
  ☆26Updated last year
• ikostrikov / jaxrl2
  ☆46Updated 2 years ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆56Updated 3 years ago
• Bellman-devs / bellman
  Model-based reinforcement learning in TensorFlow
  ☆56Updated 4 years ago
• kevinzakka / dm_env_wrappers
  Standalone library of frequently-used wrappers for dm_env environments.
  ☆17Updated last year
• ikostrikov / dmcgym
  ☆23Updated 3 years ago
• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆100Updated last year
• eager-dev / eagerx_tutorials
  Tutorials on how to use EAGERx
  ☆16Updated 2 weeks ago
• ingambe / RayEnvWrapper
  OpenAi's gym environment wrapper to vectorize them with Ray
  ☆23Updated 2 years ago
• philipjball / OffCon3
  📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)
  ☆24Updated 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
• danijar / embodied
  Fast reinforcement learning research
  ☆61Updated 8 months ago
• MinRegret / deluca
  Performant, differentiable reinforcement learning
  ☆24Updated 2 years ago
• Miffyli / policy-supervectors
  Creating fixed-length vectors to describe RL/GA policies
  ☆20Updated 3 years ago
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆82Updated 3 years ago
• toshikwa / rljax
  A collection of RL algorithms written in JAX.
  ☆103Updated 3 years ago
• RyanNavillus / reward-surfaces
  ☆17Updated last year
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆43Updated 2 years ago
• pairlab / vagram
  [ICLR 22] Value Gradient weighted Model-Based 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…
  ☆58Updated 3 years ago
• manantomar / Mirror-Descent-Policy-Optimization
  Mirror Descent Policy Optimization
  ☆38Updated 4 years ago
• frt03 / inference-based-rl
  Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)
  ☆20Updated 3 years ago
• danijar / ninjax
  General Modules for JAX
  ☆67Updated 4 months ago
• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 3 years ago
• zafarali / emdp
  Easy MDPs and grid worlds with accessible transition dynamics to do exact calculations
  ☆49Updated 3 years ago