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 2 months ago
- Modular framework for Reinforcement Learning in python☆176Updated 2 years ago
- Fully differentiable RL environments, written in Ivy.☆65Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- 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
- Baselines for gymnax 🤖☆72Updated 2 years ago
- A web based platform for collecting human actions in reinforcement learning environments☆30Updated last month
- Reinforcement learning library in JAX.☆100Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learning☆72Updated last year
- Pytrees + dataclasses ❤️☆67Updated 5 months ago
- JAX implementations of core Deep RL algorithms☆82Updated 3 years ago
- Accelerated minigrid environments with JAX☆150Updated last month
- A Python package of computer vision models for the Equinox ecosystem.☆108Updated last year
- ☆224Updated 2 years ago
- A PyTorch library for all things nonlinear control and reinforcement learning.☆47Updated 3 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆236Updated 4 months ago
- ☆83Updated last month
- A tool for recording RL trajectories.☆107Updated 2 months ago
- Tools for JAX☆49Updated 2 weeks ago
- Drop-in environment replacements that make your RL algorithm train faster.☆21Updated last year
- ☆28Updated 3 years ago
- Efficient baselines for autocurricula in JAX.☆196Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆115Updated last year
- An API conversion tool for popular external reinforcement learning environments☆190Updated last month
- ☆115Updated last month
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Updated 2 years ago
- GBRL-based Actor-Critic algorithms implemented in stable-baselines3☆38Updated last month
- Loopy belief propagation for factor graphs on discrete variables in JAX☆156Updated 11 months ago
- Library that provides environments for planning problems☆16Updated 8 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆22Updated 7 months ago