numpy>=1.15
brainunit>=0.0.8
absl-py

[cpu]
jax[cpu]
numba
brainunit

[cuda12]
jax[cuda12]
warp-lang
brainunit

[cuda13]
jax[cuda13]
warp-lang
brainunit

[testing]
pytest
jax
brainunit
absl-py

[tpu]
jax[tpu]
brainunit
