PyPI Stats

Search

All packages
Top packages

Track packages

jaxmarl


PyPI page
Home page
Author: Foerster Lab for AI Research
License: Apache License Version 2.0, January 2004 http://www.apache.org/licenses/ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 1. Definitions. ...
Summary: Multi-Agent Reinforcement Learning with JAX
Latest version: 0.2.0
Required dependencies: chex | flax | gymnax | jax | jaxtyping | matplotlib | numpy | pillow | safetensors
Optional dependencies: brax | distrax | flashbax | hydra-core | imageio | mpe2 | omegaconf | optax | pre-commit | pyright | pytest | ruff | tqdm | wandb

Downloads last day: 441
Downloads last week: 2,554
Downloads last month: 1,116
7-day smoothing