Toolbox to integrate optimal transport loss functions using automatic differentiation and Sinkhorn's algorithm - View it on GitHub