numpy
scipy
jax
jaxlib
objax
scikit-image

[opt]
matplotlib
optax
jaxopt
findiff
numpyro
blackjax
