This repository provides the necessary factory file that implements the Optimization of OT problems using Riemannian Manifold Optimization Framework. Details can be found in our work:
Bamdev Mishra, N. T. V. Satya Dev, Hiroyuki Kasai, Pratik Jawanpuria. Manifold Optimization for Optimal Transport, arXiv preprint arXiv:2103.00902, 2021.
The factory file is in compatible with the Pymanopt toolbox and can be easily integrated by placing it in the
pymanopt/manifolds/ folder (As of the Pymanopt's release
7f7a022 origin/dev ). A helper file has been provided in the
helpers directory explaining the usage of factory file.
For MATLAB implementation checkout
For GPU based acceleration (cupy), set the environment variable