issues
search
google-deepmind
/
optax
Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.63k
stars
174
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Document how to add Examples and test them in docs.
#843
copybara-service[bot]
closed
6 months ago
0
Remove multi_normal from docs (in utilities.rst) or add dosctring to multi_normal (in utils.py).
#842
vroulet
closed
1 month ago
0
Use sphinx-contributors for an automated contributors list.
#841
fabianp
closed
5 months ago
0
Avoid multiplication of boolean arrays
#840
copybara-service[bot]
closed
6 months ago
0
Add mathematical formula in docstring of linear_schedule
#839
copybara-service[bot]
closed
6 months ago
0
FIX "gallery.rst:5: WARNING: Title underline too short."
#838
copybara-service[bot]
closed
6 months ago
0
Moved linear algebra operations into utilities
#837
hmludwig
closed
6 months ago
4
Fixed the rendering of equations in jupyter notebooks
#836
hmludwig
closed
6 months ago
1
Inconsistencies in schedules API
#835
fabianp
opened
6 months ago
4
Fixed the Sharpness aware minimization docstring
#834
hmludwig
closed
6 months ago
4
Use better steop-size in doctest.
#833
copybara-service[bot]
closed
6 months ago
0
Remove misleading information bout zeroing gradients decreasing objective
#832
copybara-service[bot]
closed
6 months ago
0
Move linear algebra operations into utilities
#831
fabianp
closed
5 months ago
0
Detail SGD with momentum equations.
#830
copybara-service[bot]
closed
6 months ago
0
Math not being displayed in ogda_example.ipynb
#829
fabianp
closed
6 months ago
0
incompatibility between zero_nans() and MultiSteps
#828
TheMr33
closed
6 months ago
5
adding a perturbations module. Can take pytrees as inputs
#827
copybara-service[bot]
opened
6 months ago
0
Bad docstring formatting for SAM
#826
fabianp
closed
6 months ago
0
adding tree utils for generating trees with random values and averaging across trees.
#825
copybara-service[bot]
closed
6 months ago
0
Clarify formula for cosine learning rate.
#823
copybara-service[bot]
closed
6 months ago
0
Refactoring and link fixes in development.md
#822
fabianp
closed
6 months ago
0
adding tree utils for generating trees with random values and averaging across trees.
#821
copybara-service[bot]
closed
6 months ago
0
FIX doctest format in TransformUpdateExtraArgsFn
#820
copybara-service[bot]
closed
6 months ago
0
Experimenting with typing of OptState
#819
copybara-service[bot]
closed
6 months ago
0
Add Example, Args and Returns section to optax.linear_schedule
#818
copybara-service[bot]
closed
6 months ago
0
Hessian vector product and Gauss-Newton vector product utilities.
#817
copybara-service[bot]
opened
6 months ago
0
allow wrappers.flatten to work with 0-dim arrays
#816
copybara-service[bot]
closed
6 months ago
0
FIX: linesearch tests failing on GPU
#815
copybara-service[bot]
closed
6 months ago
0
Fixing backtracking linesearch support on gpus
#814
copybara-service[bot]
closed
6 months ago
0
Considering alternative solutions to make EmptyState a valid OptState
#813
copybara-service[bot]
closed
6 months ago
0
[FIX] bad indentation in the docstring of nadam
#812
fabianp
closed
6 months ago
0
Removed 'examples/quick_start.ipynb' and adjusted references.
#811
hmludwig
closed
6 months ago
0
[DOC] streamline sidebar of gallery
#810
fabianp
closed
6 months ago
0
remove quickstart example (https://github.com/google-deepmind/optax/blob/main/examples/quick_start.ipynb), now that it has been merged into optax-101.ipynb
#809
fabianp
closed
6 months ago
2
Unsupported operand type for float and dict while updating params
#808
ArunDtej
closed
6 months ago
2
Benchmark recommendations.
#807
copybara-service[bot]
closed
6 months ago
0
Fix RST formatting in schedules docstrings
#806
copybara-service[bot]
closed
6 months ago
0
Implementation of Eve Optimizer
#805
amosyou
opened
6 months ago
1
Adjusted 'test.sh' for universal macOS and Linux Compatibility
#804
hmludwig
closed
6 months ago
1
Fix NamedTuples Formatting
#803
amosyou
opened
6 months ago
8
Make EmptyState a valid ArrayTree
#802
copybara-service[bot]
closed
6 months ago
0
Make optax.EmptyState a valid optax.OptState
#801
copybara-service[bot]
closed
6 months ago
0
Add tree_map_params to the API documentation
#800
copybara-service[bot]
closed
6 months ago
0
internal change
#799
copybara-service[bot]
closed
6 months ago
0
make EmptyState its own class
#798
copybara-service[bot]
closed
6 months ago
0
Improve error message for `cosine_decay_schedule`.
#797
copybara-service[bot]
closed
6 months ago
0
Removing experimental api for now.
#796
copybara-service[bot]
closed
6 months ago
0
Backtracking linesearch.
#795
copybara-service[bot]
closed
6 months ago
0
Add missing examples to contrib examples section.
#794
fabianp
closed
6 months ago
0
Fixing the update line, updating all test cases
#793
copybara-service[bot]
closed
6 months ago
0
Previous
Next