jax
flax
gymnasium
matplotlib
seaborn

[dev]
pytest
pytest-cov
ruff
mypy

[test]
chex
flax
brax
minatar
