GradNetOT

GradNetOT: Learning Optimal Transport Maps with GradNets

GradNetOT learns optimal transport maps as the gradient of a convex potential, using a monotone gradient network trained on the Monge–Ampère residual.