Farama-Foundation / JumpyLinks
On-the-fly conversions between Jax and NumPy tensors
☆52Updated 2 years ago
Alternatives and similar repositories for Jumpy
Users that are interested in Jumpy are comparing it to the libraries listed below
Sorting:
- Performant, differentiable reinforcement learning☆123Updated 3 weeks ago
- Fully differentiable RL environments, written in Ivy.☆65Updated 2 years ago
- Modular framework for Reinforcement Learning in python☆175Updated 2 years ago
- A web based platform for collecting human actions in reinforcement learning environments☆31Updated 2 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆73Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Reinforcement learning library in JAX.☆100Updated last year
- Baselines for gymnax 🤖☆71Updated 2 years ago
- ☆222Updated last year
- Pytrees + dataclasses ❤️☆68Updated 4 months ago
- Accelerated minigrid environments with JAX☆144Updated 2 weeks ago
- A Python package of computer vision models for the Equinox ecosystem.☆108Updated last year
- An API conversion tool for popular external reinforcement learning environments☆175Updated 2 weeks ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆232Updated 3 months ago
- ☆82Updated 5 months ago
- Drop-in environment replacements that make your RL algorithm train faster.☆21Updated last year
- JAX implementations of core Deep RL algorithms☆82Updated 3 years ago
- Plug-and-play hydra sweepers for the EA-based multifidelity method DEHB and several population-based training variations, all proven to e…☆83Updated last year
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆83Updated last year
- A tool for recording RL trajectories.☆107Updated last month
- A Fast, Portable Deep Reinforcement Learning Library for Continuous Control☆14Updated 2 years ago
- ☆115Updated last week
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆22Updated 6 months ago
- ☆15Updated last year
- Loopy belief propagation for factor graphs on discrete variables in JAX☆154Updated 10 months ago
- JAX implementation of RL algorithms and vectorized environments☆47Updated last year
- Building blocks for productive research☆59Updated 3 weeks ago
- GBRL-based Actor-Critic algorithms implemented in stable-baselines3☆38Updated 2 weeks ago
- Efficient baselines for autocurricula in JAX.☆196Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆58Updated 3 years ago