numpy

[all]
torch
jax
requests

[dev]
pytest
pytest-mock
black
flake8
build
twine

[fetch]
requests

[jax]
jax

[jax_cuda12]
jax[cuda12]

[jax_tpu]
jax[tpu]

[torch]
torch
