jax
jaxlib
matplotlib
nibabel
numba
numpy
scikit-learn
