numpy>=1.20 scipy>=1.3 matplotlib autograd pymanopt cvxopt scikit-learn torch jax jaxlib tensorflow pytest