jax
jaxlib
nibabel
numba
numpy
sklearn
