Farama-Foundation / JumpyLinks
On-the-fly conversions between Jax and NumPy tensors
☆57Updated 2 years ago
Alternatives and similar repositories for Jumpy
Users that are interested in Jumpy are comparing it to the libraries listed below
Sorting:
- Modular framework for Reinforcement Learning in python☆181Updated 2 years ago
- Performant, differentiable reinforcement learning☆125Updated 5 months ago
- Fully differentiable RL environments, written in Ivy.☆66Updated 2 years ago
- Reinforcement learning library in JAX.☆100Updated 2 years ago
- A web based platform for collecting human actions in reinforcement learning environments☆31Updated 4 months ago
- Baselines for gymnax 🤖☆74Updated 2 years ago
- Accelerated minigrid environments with JAX☆154Updated 2 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆74Updated last year
- ☆91Updated 4 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- 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…☆85Updated 2 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆254Updated 2 months ago
- Drop-in environment replacements that make your RL algorithm train faster.☆21Updated last year
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆43Updated 3 years ago
- An API conversion tool for popular external reinforcement learning environments☆199Updated last month
- GBRL-based Actor-Critic algorithms implemented in stable-baselines3☆42Updated last week
- An Open-Ended Agentic Simulator☆58Updated last year
- JAX implementation of RL algorithms and vectorized environments☆51Updated 2 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Updated 11 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- A tool for recording RL trajectories.☆111Updated 5 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆59Updated 3 years ago
- Accelerated replay buffers in JAX☆46Updated 3 years ago
- Pytrees + dataclasses ❤️☆71Updated 3 weeks ago
- Highly scalable 2D JAX physics engine.☆61Updated 10 months ago
- Fast and procedurally generated side-scroller-game-like graphical environments (formerly Procgen)☆32Updated 2 years ago
- Reinforcement learning training framework for entity-gym environments.☆17Updated last year