torch>=2.0.0
torchvision>=0.15.0
numpy
scipy
scikit-learn
matplotlib
geomloss>=0.2.6
POT>=0.9.0
tqdm
pyyaml
# Optional: pykeops for GPU-accelerated Sinkhorn
# pykeops