/ssax

Implementation of Sinkhorn Step in JAX, NeurIPS 2023.

Primary LanguagePythonMIT LicenseMIT

This repository is not active