numpy>=1.20
scipy>=1.6
matplotlib
autograd
pymanopt
cvxopt
scikit-learn
torch<2.12
jax
jaxlib
tensorflow; python_version < '3.14'
pytest
torch_geometric
cvxpy
geomloss>=0.3.1
pykeops
