Skip to content

Pull requests: google-deepmind/optax

New pull request New
22 Open 790 Closed
22 Open 790 Closed
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
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Extend example gallery entry for linear assignment problem with optimal transport example.
#1143 opened Nov 23, 2024 by carlosgmartin Loading…
Fix complex support for L-BFGS
#1142 opened Nov 22, 2024 by gautierronan Loading…
1
2
Add improved version of Hungarian algorithm.
#1140 opened Nov 20, 2024 by carlosgmartin Loading…
4
Replaced seed with key in add_noise and noisy_sgd
#1138 opened Nov 13, 2024 by Tomas542 Loading…
13
Add Muon Optimizer to contrib
#1126 opened Nov 2, 2024 by leloykun Loading…
4
[Triplet Margin Loss] Issue 1118
#1120 opened Oct 26, 2024 by cvnad1 Loading…
26
Changes to test.sh and pyproject.toml.
#1110 opened Oct 17, 2024 by carlosgmartin Loading…
15
Adds AdeMAMix Optimizer to contrib
#1104 opened Oct 14, 2024 by mathDR Loading…
1
49
Adding notebook illustrating schedule_free optimizer integration
#1022 opened Jul 24, 2024 by copybara-service bot Loading…
Add sophia-h optimizer
#979 opened May 29, 2024 by evanatyourservice Loading…
41
Gauss-Newton and Levenberg-Marquardt
#920 opened Apr 9, 2024 by gbruno16 • Draft
@vroulet
Simplification for transforms with empty state
#895 opened Mar 28, 2024 by fabianp Loading…
1
Fix init_value and end_value in cosine decay
#884 opened Mar 26, 2024 by copybara-service bot Loading…
Test on Python 3.12
#862 opened Mar 12, 2024 by fabianp Loading…
1
Hessian vector product and Gauss-Newton vector product utilities.
#817 opened Feb 21, 2024 by copybara-service bot Loading…
Implementation of Eve Optimizer
#805 opened Feb 15, 2024 by amosyou • Draft
1
Fix NamedTuples Formatting
#803 opened Feb 15, 2024 by amosyou Loading…
15
Remove TF dependency in adversarial_training example
#780 opened Feb 7, 2024 by mmhamdy Loading…
2
Remove TF dependency in reduce_on_plateau example
#768 opened Feb 4, 2024 by mmhamdy Loading…
13
Allow gradient transform parameters to be dynamic
#516 opened Mar 22, 2023 by NeilGirdhar Loading…
7
Adds Eve Optimizer needs work Needs more work by the author to be merged
#475 opened Jan 24, 2023 by wglao Loading…
4
Add example of reading learning rate from optimizer state
#363 opened Jun 18, 2022 by atgctg Loading…
1
4
ProTip! Add no:assignee to see everything that’s not assigned.