numpy>=1.9
scipy>=1.0
autograd>=1.4
colorlog

[:python_version < "3.8"]
importlib_metadata

[all]
torch
jaxlib
jax[cuda]
scikit-cuda

[gpu]
jax[cuda]

[jax]
jax
jaxlib

[magma]
scikit-cuda

[torch]
torch
