numpy>=1.15
brainunit>=0.3.0
brainevent>=0.0.7

[cpu]
jax[cpu]>=0.7.0
brainunit>=0.3.0
brainevent>=0.0.7

[cuda12]
jax[cuda12]>=0.7.0
brainunit>=0.3.0
brainevent>=0.0.7

[cuda13]
jax[cuda13]>=0.7.0
brainunit>=0.3.0
brainevent>=0.0.7

[progressbar]
tqdm

[testing]
absl-py
pytest
coverage
jax>=0.7.0
tqdm
brainunit>=0.3.0
brainevent>=0.0.7

[tpu]
jax[tpu]>=0.7.0
brainunit>=0.3.0
brainevent>=0.0.7
