jax
gymnasium
matplotlib
seaborn

[dev]
pytest
pytest-cov
ruff
mypy

[test]
chex
minatar
absl-py
