/ssax

Implementation of Sinkhorn Step in JAX, NeurIPS 2023.

Primary LanguagePythonMIT LicenseMIT

Stargazers