google-deepmind optax issues

google-deepmind / optax

Optax is a gradient processing and optimization library for JAX.

https://optax.readthedocs.io

Apache License 2.0

1.57k stars 169 forks source link

issues

Newest

Newest Most commented Recently updated Oldest Least commented Least recently updated

FIX RST formatting in inject hyperparams

#867 fabianp closed 4 months ago
0
[DOC] Add to the gallery an example on a small language model

#866 copybara-service[bot] closed 3 months ago
0
Utility to set value in a pytree (and so in state)

#865 copybara-service[bot] closed 4 months ago
0
Document the extra args of the update function in docstring

#864 copybara-service[bot] closed 4 months ago
0
Warn that in future arguments after the initial (prediction, ground_truth) positional arguments will become keyword-only in optax losses.

#863 copybara-service[bot] closed 4 months ago
0
Test on Python 3.12

#862 fabianp opened 4 months ago
1
[IGNORE] dummy commit to see if RDT build upon pull requests

#861 fabianp closed 4 months ago
0
Error in the lookahead optimizer example

#860 fabianp closed 3 months ago
0
Utility to extract value from pytree (and so from state)

#859 copybara-service[bot] closed 4 months ago
0
ENH: extend power_iteration to accept a matrix in implicit form

#858 copybara-service[bot] closed 3 months ago
0
Added mathematical description to Noisy SGD

#857 hmludwig closed 4 months ago
1
start of 0.2.2 development cycle

#856 copybara-service[bot] closed 4 months ago
0
0.2.1 release

#855 copybara-service[bot] closed 4 months ago
0
Broken tarball for release `v0.2.0`

#854 daskol closed 4 months ago
3
Fix sign discrepancy in tolerance in scale_by_backtracking_linesearch

#853 copybara-service[bot] closed 4 months ago
0
Document in docstring recent API changes in _tree_math.py

#852 copybara-service[bot] closed 4 months ago
0
Reference reduce on plateau example from docstring

#851 copybara-service[bot] closed 4 months ago
0
Break the scale_by_backtracking_linesearch docstring example for better readability

#850 copybara-service[bot] closed 4 months ago
0
"dadapt_adamw, combine, prodigy" occur "TypeError: 'type' object is not subscriptable."

#849 sbb2002 closed 4 months ago
1
typo power_iteratio -> power_iteration

#847 copybara-service[bot] closed 4 months ago
0
Clarity improvements for scale_by_backtracking_linesearch docstring

#846 copybara-service[bot] closed 4 months ago
0
Begin of 0.2.1 development

#845 copybara-service[bot] closed 4 months ago
0
v0.2.0 release

#844 copybara-service[bot] closed 4 months ago
0
Document how to add Examples and test them in docs.

#843 copybara-service[bot] closed 4 months ago
0
Remove multi_normal from docs (in utilities.rst) or add dosctring to multi_normal (in utils.py).

#842 vroulet closed 5 days ago
0
Use sphinx-contributors for an automated contributors list.

#841 fabianp closed 4 months ago
0
Avoid multiplication of boolean arrays

#840 copybara-service[bot] closed 4 months ago
0
Add mathematical formula in docstring of linear_schedule

#839 copybara-service[bot] closed 4 months ago
0
FIX "gallery.rst:5: WARNING: Title underline too short."

#838 copybara-service[bot] closed 4 months ago
0
Moved linear algebra operations into utilities

#837 hmludwig closed 4 months ago
4
Fixed the rendering of equations in jupyter notebooks

#836 hmludwig closed 4 months ago
1
Inconsistencies in schedules API

#835 fabianp opened 4 months ago
4
Fixed the Sharpness aware minimization docstring

#834 hmludwig closed 4 months ago
4
Use better steop-size in doctest.

#833 copybara-service[bot] closed 4 months ago
0
Remove misleading information bout zeroing gradients decreasing objective

#832 copybara-service[bot] closed 4 months ago
0
Move linear algebra operations into utilities

#831 fabianp closed 4 months ago
0
Detail SGD with momentum equations.

#830 copybara-service[bot] closed 4 months ago
0
Math not being displayed in ogda_example.ipynb

#829 fabianp closed 4 months ago
0
incompatibility between zero_nans() and MultiSteps

#828 TheMr33 closed 4 months ago
5
adding a perturbations module. Can take pytrees as inputs

#827 copybara-service[bot] opened 4 months ago
0
Bad docstring formatting for SAM

#826 fabianp closed 4 months ago
0
adding tree utils for generating trees with random values and averaging across trees.

#825 copybara-service[bot] closed 4 months ago
0
Clarify formula for cosine learning rate.

#823 copybara-service[bot] closed 4 months ago
0
Refactoring and link fixes in development.md

#822 fabianp closed 4 months ago
0
adding tree utils for generating trees with random values and averaging across trees.

#821 copybara-service[bot] closed 4 months ago
0
FIX doctest format in TransformUpdateExtraArgsFn

#820 copybara-service[bot] closed 4 months ago
0
Experimenting with typing of OptState

#819 copybara-service[bot] closed 4 months ago
0
Add Example, Args and Returns section to optax.linear_schedule

#818 copybara-service[bot] closed 4 months ago
0
Hessian vector product and Gauss-Newton vector product utilities.

#817 copybara-service[bot] opened 5 months ago
0
allow wrappers.flatten to work with 0-dim arrays

#816 copybara-service[bot] closed 5 months ago
0

Previous Next