Farama-Foundation / JumpyLinks
On-the-fly conversions between Jax and NumPy tensors
☆53Updated 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☆178Updated 2 years ago
- Fully differentiable RL environments, written in Ivy.☆65Updated 2 years ago
- Performant, differentiable reinforcement learning☆123Updated 2 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Baselines for gymnax 🤖☆72Updated 2 years ago
- A web based platform for collecting human actions in reinforcement learning environments☆30Updated last month
- Contains JAX implementation of algorithms for inverse reinforcement learning☆72Updated last year
- Plug-and-play hydra sweepers for the EA-based multifidelity method DEHB and several population-based training variations, all proven to e…☆84Updated last year
- ☆85Updated last month
- Reinforcement learning library in JAX.☆100Updated 2 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆243Updated last week
- JAX implementations of core Deep RL algorithms☆82Updated 3 years ago
- Accelerated minigrid environments with JAX☆151Updated last week
- Efficient baselines for autocurricula in JAX.☆197Updated last year
- Drop-in environment replacements that make your RL algorithm train faster.☆21Updated last year
- A tool for recording RL trajectories.☆108Updated 3 months ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆59Updated 3 years ago
- Pytrees + dataclasses ❤️☆67Updated 6 months ago
- GBRL-based Actor-Critic algorithms implemented in stable-baselines3☆38Updated last week
- A PyTorch library for all things nonlinear control and reinforcement learning.☆47Updated 4 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆38Updated 3 weeks ago
- ☆15Updated 2 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆116Updated last year
- Image augmentation library for Jax☆40Updated last year
- An API conversion tool for popular external reinforcement learning environments☆191Updated this week
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆22Updated 8 months ago
- ☆115Updated last month
- ☆28Updated 3 years ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago