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

Fix formatting issues with gallery entry for linear assignment problem. #1095

Closed carlosgmartin closed 1 month ago

carlosgmartin commented 1 month ago

Looks like https://optax.readthedocs.io/en/latest/_collections/examples/linear_assignment_problem.html has some formatting issues. This PR should fix it.