numpy>=1.20.0

[all]
jax[cpu]
jaxlib

[all:platform_machine == "x86_64"]
cupy-cuda12x

[dev]
pytest>=6.0
pytest-cov
black
flake8
mypy

[gpu]

[gpu:platform_machine == "x86_64"]
cupy-cuda12x

[jax]
jax[cpu]
jaxlib
