numpy>=1.15
tqdm
brainpy
brainpylib
numba
mpi4py
mpi4jax
jax[cpu]==0.3.24
