numpy>=1.15
jax>=0.2.10
tqdm

[cpu]
jaxlib>=0.1.64
brainpylib>=0.03

[cuda]
jaxlib>=0.1.64
brainpylib>=0.03
