Pytorch implementation of "Neural Optimal Transport with General Cost Functionals" (ICLR 2024)
Primary LanguageJupyter Notebook