numpy

[all]
torch
jax
requests
numba

[dev]
pytest
pytest-mock
black
flake8
build
twine

[fast]
numba

[fetch]
requests

[jax]
jax

[jax_cuda12]
jax[cuda12]

[jax_tpu]
jax[tpu]

[torch]
torch
