Optimization using the Python Optimal Transport Package