numpy>=1.15
tqdm
brainunit
brainevent

[cpu]
jax[cpu]
brainunit
brainevent

[cuda12]
jax[cuda12]
brainunit
brainevent

[cuda13]
jax[cuda13]
brainunit
brainevent

[testing]
absl-py
pytest
jax
brainunit
brainevent

[tpu]
jax[tpu]
brainunit
brainevent
