anindex / ssax

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

Related projects

Alternatives and complementary repositories for ssax