James Thornton

Minor, I assume best practice is to import the module not the class or function https://github.com/ott-jax/ott/blob/5657e786c99afb808edd98e061ec316eed0942be/ott/core/icnn.py#L26

Thank you! This works, apologies if it's in the docs and I missed it. Also, there is an LBFGS optimizer in tfp (does not require tensorflow itself) and works well...

Hello, I am also interested in the Jax version

This code is no longer being maintained. I will release an improved Jax version later this year.