flash_sinkhorn
