anindex / ssax

Implementation of Sinkhorn Step in JAX, NeurIPS 2023.
41Updated last month

Related projects

Alternatives and complementary repositories for ssax