numpy>=1.15
tqdm
brainpy==2.2.4.0
numba
mpi4py==3.1.4
mpi4jax==0.3.11
jax==0.3.25
matplotlib
