numpy
matplotlib
scipy
tqdm

[git]
GitPython

[jax]

[jax:sys_platform != "darwin"]
jax[cuda]

[jax:sys_platform == "darwin"]
jax[cpu]

[mpi]
mpi4py

[opencl]
pyopencl
mako
