Code corresponding to the paper Diffusion Earth Mover's Distance and Distribution Embeddings - View it on GitHub