Hotfix: numercial stability of non-log-stabilized sinkhorn plan#531

Merged

LarsKue merged 16 commits intodevfrom

fix-sinkhorn-plan

Jul 8, 2025

Commits on Jul 5, 2025

sinkhorn_plan now returns a transport plan with uniform marginal distributions
daniel-habermann
committed
add unit test for sinkhorn_plan
daniel-habermann
committed
fix sinkhorn function by sampling from the logits of the transpose of the plan, instead of the plan directly
daniel-habermann
committed
sinkhorn(x1, x2) now samples from log(plan) to receive assignments such that x2[assignments] matches x1
daniel-habermann
committed
re-enable test_assignment_is_optimal() for method='sinkhorn'
daniel-habermann
committed
log_sinkhorn now correctly uses log_plan instead of keras.ops.exp(log_plan), log_sinkhorn_plan returns logits of the transport plan
daniel-habermann
committed
add unit tests for log_sinkhorn_plan
daniel-habermann
committed
fix faulty indexing with tensor for tensorflow backend
daniel-habermann
committed

Commits on Jul 8, 2025

re-add numItermax for ot pot test
LarsKue
committed