| 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 | |
| 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 | |