jax
jaxlib
matplotlib
numpy
psutil
scikit-learn
sklearn
