search

google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.71k stars 194 forks source link

Extend example gallery entry for linear assignment problem with optimal transport example. #1143

Open carlosgmartin opened 4 days ago

carlosgmartin commented 4 days ago

Follow-up to https://github.com/google-deepmind/optax/pull/1083#issuecomment-2407808657.

See also https://github.com/ott-jax/ott/pull/598.

carlosgmartin commented 22 hours ago

@vroulet Just noting that ott-jax is now using optax's Hungarian algorithm for unregularized OT solving.

vroulet commented 22 hours ago

Very nice!