Hi @AjayTalati,
I started an implementation here: https://github.com/t-vi/pytorch-tvmisc/blob/master/wasserstein-distance/Pytorch_Wasserstein.ipynb 319
I’m not terribly impressed by the numerical stability, I’ll have to look into that.
Best regards
Thomas