Skip to content
Permalink

Comparing changes

Choose two branches to see what’s changed or to start a new pull request. If you need to, you can also or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: google-deepmind/optax
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: v0.2.0
Choose a base ref
...
head repository: google-deepmind/optax
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: v0.2.1
Choose a head ref
  • 10 commits
  • 11 files changed
  • 2 contributors

Commits on Mar 5, 2024

  1. Begin of 0.2.1 development 

    PiperOrigin-RevId: 612951829
    @vroulet
    vroulet authored and OptaxDev committed Mar 5, 2024
    Loading
    Loading status checks…
    Copy the full SHA
    1b93a43 View commit details

Commits on Mar 6, 2024

  1. Add mathematical formula in docstring of linear_schedule 

    For consistency I also changed notation \gamma_0 -> I in `cosine_decay_schedule`.

PiperOrigin-RevId: 613128067
    @fabianp
    fabianp authored and OptaxDev committed Mar 6, 2024
    Loading
    Loading status checks…
    Copy the full SHA
    fa267b4 View commit details

  2. typo power_iteratio -> power_iteration 

    PiperOrigin-RevId: 613128962
    @fabianp
    fabianp authored and OptaxDev committed Mar 6, 2024
    Copy the full SHA
    627a08a View commit details

  3. Clarity improvements for scale_by_backtracking_linesearch docstring 

    PiperOrigin-RevId: 613195688
    @fabianp
    fabianp authored and OptaxDev committed Mar 6, 2024
    Loading
    Loading status checks…
    Copy the full SHA
    7f63013 View commit details

Commits on Mar 7, 2024

  1. Document in docstring recent API changes in _tree_math.py 

    PiperOrigin-RevId: 613566096
    @fabianp
    fabianp authored and OptaxDev committed Mar 7, 2024
    Loading
    Loading status checks…
    Copy the full SHA
    c7aac6d View commit details

  2. Break the scale_by_backtracking_linesearch docstring example for bett… 

    …er readability

Now it displays into two code blocks, which I think is more readable.

Also removed a dangling mention to api/linear_algebra_operators, file
that no longer exists

PiperOrigin-RevId: 613581996
    @fabianp
    fabianp authored and OptaxDev committed Mar 7, 2024
    Loading
    Loading status checks…
    Copy the full SHA
    16f1ee9 View commit details

  3. Reference reduce on plateau example from docstring 

    Other misc changes:

   * Changed title of reduce_on_plateau.ipynb example to emphasize the method instead of the dataset.
   * replaced deprecated jupyter_execute_notebooks by nb_execution_mode in Makefile command line

PiperOrigin-RevId: 613582429
    @fabianp
    fabianp authored and OptaxDev committed Mar 7, 2024
    Loading
    Loading status checks…
    Copy the full SHA
    8700727 View commit details

  4. Fix sign discrepancy in tolerance in scale_by_backtracking_linesearch 

    the tolerance should be (1 + \delta) instead of (1 - \delta)

PiperOrigin-RevId: 613613682
    @fabianp
    fabianp authored and OptaxDev committed Mar 7, 2024
    Loading
    Loading status checks…
    Copy the full SHA
    de46234 View commit details

  5. Document how to add Examples and test them in docs. 

    PiperOrigin-RevId: 613615583
    @vroulet
    vroulet authored and OptaxDev committed Mar 7, 2024
    Loading
    Loading status checks…
    Copy the full SHA
    18110a4 View commit details

  6. 0.2.1 release 

    PiperOrigin-RevId: 613637111
    @fabianp
    fabianp authored and OptaxDev committed Mar 7, 2024
    Loading
    Loading status checks…
    Copy the full SHA
    5f2a6fe View commit details
Loading