jax
jaxopt
scikit-learn
