numpy
optax
torch
jax

[doc]
sphinx
sphinx_rtd_theme

[tests]
pytest
pytest_lazy_fixture
