Alternatives and similar repositories for Jumpy

Users that are interested in Jumpy are comparing it to the libraries listed below

• coax-dev / coax
  Modular framework for Reinforcement Learning in python
  ☆182Updated 3 years ago
• google / deluca
  Performant, differentiable reinforcement learning
  ☆125Updated 6 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• RobertTLange / gymnax-blines
  Baselines for gymnax 🤖
  ☆74Updated 2 years ago
• ivy-llc / gym
  Fully differentiable RL environments, written in Ivy.
  ☆66Updated 2 years ago
• Farama-Foundation / CrowdPlay
  A web based platform for collecting human actions in reinforcement learning environments
  ☆31Updated 4 months ago
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆83Updated 3 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆74Updated last month
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆40Updated 2 years ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆110Updated last year
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆74Updated last year
• google-deepmind / tf2jax
  ☆120Updated last week
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆156Updated 3 months ago
• ethanluoyc / magi
  Reinforcement learning library in JAX.
  ☆100Updated 2 years ago
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆260Updated 3 months ago
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆164Updated last year
• DramaCow / jaxued
  ☆91Updated 2 weeks ago
• facebookresearch / how-to-autorl
  Plug-and-play hydra sweepers for the EA-based multifidelity method DEHB and several population-based training variations, all proven to e…
  ☆85Updated 2 years ago
• kenjyoung / mctx_learning_demo
  ☆53Updated 2 years ago
• Farama-Foundation / Procgen2
  Fast and procedurally generated side-scroller-game-like graphical environments (formerly Procgen)
  ☆33Updated 2 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆41Updated last year
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year
• NVlabs / gbrl_sb3
  GBRL-based Actor-Critic algorithms implemented in stable-baselines3
  ☆43Updated last week
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆11Updated 5 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆49Updated 2 weeks ago
• RPegoud / jym
  JAX implementation of RL algorithms and vectorized environments
  ☆51Updated 2 years ago
• google / trajax
  ☆233Updated 2 years ago
• Farama-Foundation / Shimmy
  An API conversion tool for popular external reinforcement learning environments
  ☆201Updated last month
• google-deepmind / envlogger
  A tool for recording RL trajectories.
  ☆111Updated 6 months ago
• sash-a / CleanRL.jl
  Simple single file implementations of Reinforcement Learning algorithms in Julia
  ☆23Updated 11 months ago