Implement functionality of the GeomLoss python package
This PhD thesis does a fantastic job of describing various optimization techniques for the GPU implementation of Sinkhorn. It should be straightforward to include several of these changes.
This PhD thesis does a fantastic job of describing various optimization techniques for the GPU implementation of Sinkhorn. It should be straightforward to include several of these changes.