numpy
scipy
pandas
jax
flax
optax
orbax-checkpoint

[docs]
sphinx
furo

[lint]
ruff

[test]
pytest
pytest-cov
