numpy==1.26.4
scikit-learn==1.6.1
POT==0.9.5
jax==0.5.2
ott-jax==0.5.0
