absl-py
bsuite
black
chex
distrax
dm-haiku
dm-tree
evosax
flax
hydra-core
isort
jax
jaxlib
numpy
optax
pytest
rlax
wandb
pytest
