Skip to content

Pull requests: google-deepmind/optax

Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Assignee
Filter by who’s assigned
Sort

Pull requests list

Adds Eve Optimizer needs work Needs more work by the author to be merged
#475 opened Jan 24, 2023 by wglao Review required
Remove TF dependency in reduce_on_plateau example
#768 opened Feb 4, 2024 by mmhamdy Loading…
Fix NamedTuples Formatting
#803 opened Feb 15, 2024 by amosyou Loading…
Implementation of Eve Optimizer
#805 opened Feb 15, 2024 by amosyou Draft
Test on Python 3.12
#862 opened Mar 12, 2024 by fabianp Loading…
Fix init_value and end_value in cosine decay
#884 opened Mar 26, 2024 by copybara-service bot Loading…
Simplification for transforms with empty state
#895 opened Mar 28, 2024 by fabianp Loading…
Update references to JAX's GitHub repo
#1151 opened Dec 5, 2024 by copybara-service bot Loading…
replaced seed with key
#1167 opened Jan 6, 2025 by Tomas542 Loading…
Stop gradient in linesearch identity scaling
#1190 opened Feb 5, 2025 by younik Loading…
fix: Unexpected tracer in muon optimizer when sharding.
#1193 opened Feb 13, 2025 by hlzl Loading…
Fix nnls to handle batch shapes correctly.
#1200 opened Feb 17, 2025 by carlosgmartin Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.