Learning with a Wasserstein Loss Hossein Mobahi Center for Brains, Minds and Machines