numpy>=1.20 scipy>=1.3 matplotlib autograd pymanopt==0.2.4; python_version <'3' pymanopt==0.2.6rc1; python_version >= '3' cvxopt scikit-learn torch jax jaxlib tensorflow pytest