File size: 167 Bytes
4b9c417 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | 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 |
4b9c417 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | 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 |