/scalable-pytorch-sinkhorn

Fast, Memory-Efficient Approximate Wasserstein Distances. This repository contains PyTorch code to compute fast p-Wasserstein distances between d-dimensional point clouds using the Sinkhorn Algorithm. This implementation uses linear memory overhead and is stable in float32, runs on the GPU, and fully differentiable.

Primary LanguagePython

Watchers