A common task in machine learning is measuring the distance between two probability distributions. For example, suppose distribution ... move_dirt() as nested local functions inside the my_wasserstein ...