numpy>=1.26
scipy>=1.11

[all]
torch
tensorflow
jax
scikit-learn

[dev]
pytest
build
twine

[jax]
jax

[sklearn]
scikit-learn

[tensorflow]
tensorflow

[torch]
torch
