Transportation

Wasserstein-2 Optimal Transport (w2ot) w2ot is JAX software by Brandon Amos for estimating Wasserstein-2 optimal transport maps between continuous measures in Euclidean space. This is the official source code behind the paper on