jax
jaxlib
jupyter
matplotlib
scikit-learn
