Differentiable OT with PyTorch
------------------------------