This comes recommended for pytorch https://github.com/jeanfeydy/geomloss 179
A more mature and comprehensive implementation of all things OT related is https://github.com/rflamary/POT 83, however they do not yet have pytorch support. That on their todo list