numpy
pandas
matplotlib
chromax
gymnasium>=0.29
jax
jaxtyping

[docs]
sphinx-rtd-theme

[testing]
pytest
