numpy>=1.15
tqdm
brainunit
brainevent

[cpu]
jax[cpu]>=0.6.0
brainunit
brainevent

[cuda12]
jax[cuda12]>=0.6.0
brainunit
brainevent

[cuda13]
jax[cuda13]>=0.6.0
brainunit
brainevent

[testing]
absl-py
pytest
jax>=0.6.0
brainunit
brainevent

[tpu]
jax[tpu]>=0.6.0
brainunit
brainevent
