Skip to content
This repository has been archived by the owner on Aug 31, 2022. It is now read-only.

new tests to check jacobian, small fixes in: transport interface, handling of kernel_ridge in implicit differentiation of sinkhorn

Compare
Choose a tag to compare
@marcocuturi marcocuturi released this 23 Jul 15:10
· 133 commits to master since this release

In this release:

  • more tests to check accuracy of differentiation of the optimal potentials w.r.t. input parameters.
  • fixes issues in transport.Transport handling of **kwargs, that must be routed to the epsilon_scheduler object.
  • cancel ridge_kernel in the unbalanced case when doing implicit differentiation.