POT
POT copied to clipboard
Implement set_gradient for JAX backend
One can implement it easily by using compensating sums.