23 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Kaggle Environments | 87K | |
| (Crafter + NetHack) in JAX. ICML 2024 Spotlight. | 7K | |
| Multi-Agent Reinforcement Learning with JAX | 6K | |
| Cost constrained environments with a gymnax interface | 4K | |
| A continual reinforcement learning benchmark | 4K | |
| JAX implementation of Reinforcement Learning agents | 1K | |
| Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral... | 850 | |
| Interoperating between (Deep) Reiforcement Learning libraries | 809 | |
| A reinforcement learning environment for simulating target-MDP environments (Air... | 796 | |
| Lux AI Challenge Season 3 environment code | 794 | |
| Vectorizable RL algorithms in pure JAX | 637 | |
| Add your description here | 427 | |
| LEM slope stability analysis using JAX | 334 | |
| A simple and useable UED library in Jax | 320 | |
| TargetGym: Fast Setpoint tracking environments in Jax. | 273 | |
| Bayesian Data Analysis using PyMC and JAX | 255 | |
| A unified JAX/Flax framework for memory-augmented reinforcement learning with su... | 219 | |
| Python Boilerplate that contains all the code you need to create a Python packag... | 206 | |
| Reinforcement learning in pure JAX. | 151 | |
| Automatically created RL Environment Editors | 144 | |
| Interoperate among reinforcement learning libraries with jax, pytorch, gym and d... | 115 | |
| Partially Observable Benchmarks in JAX | 90 | |
| Synthetic gymnax environments | 69 |
