Farama-Foundation / Jumpy
On-the-fly conversions between Jax and NumPy tensors
ā51Updated 2 years ago
Alternatives and similar repositories for Jumpy:
Users that are interested in Jumpy are comparing it to the libraries listed below
- Modular framework for Reinforcement Learning in pythonā172Updated 2 years ago
- Baselines for gymnax š¤ā66Updated 2 years ago
- JAX implementation of RL algorithms and vectorized environmentsā43Updated last year
- Performant, differentiable reinforcement learningā120Updated 10 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learningā72Updated 8 months ago
- Accelerated minigrid environments with JAXā134Updated 9 months ago
- Plug-and-play hydra sweepers for the EA-based multifidelity method DEHB and several population-based training variations, all proven to eā¦ā78Updated last year
- Fully differentiable RL environments, written in Ivy.ā64Updated last year
- ā76Updated last month
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!ā205Updated last month
- JAX implementations of core Deep RL algorithmsā79Updated 3 years ago
- A web based platform for collecting human actions in reinforcement learning environmentsā28Updated last year
- An API conversion tool for popular external reinforcement learning environmentsā162Updated 3 months ago
- ā28Updated 2 years ago
- Highly scalable 2D JAX physics engine.ā55Updated last month
- ā75Updated 6 months ago
- A small library for creating and manipulating custom JAX Pytree classesā56Updated 2 years ago
- A tool for recording RL trajectories.ā101Updated 5 months ago
- Reinforcement learning in pure JAX.ā12Updated 2 months ago
- ā” Flashbax: Accelerated Replay Buffers in JAXā234Updated last month
- Image augmentation library for Jaxā39Updated last year
- Benchmarks for Multi-Objective Multi-Agent Decision Makingā87Updated last month
- Accelerated replay buffers in JAXā41Updated 2 years ago
- Standard interface for entity based reinforcement learning environments.ā37Updated last year
- A project that provides help for using DeepMind's mctx on gym-style environments.ā59Updated 5 months ago
- Drop-in environment replacements that make your RL algorithm train faster.ā20Updated 10 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)ā10Updated last year
- Generative cellular automaton-like learning environments for RL.ā19Updated 3 months ago
- Reinforcement learning library in JAX.ā100Updated last year
- Fast and procedurally generated side-scroller-game-like graphical environments (formerly Procgen)ā28Updated last year