Skip to content

Pull requests: google-deepmind/optax

New pull request New
25 Open 983 Closed
25 Open 983 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Expose scale by muon mask in the muon alias
#1407 opened Aug 22, 2025 by copybara-service bot Loading…
Expose scale by muon mask in the muon alias
#1402 opened Aug 12, 2025 by copybara-service bot Loading…
Add missing equations to optax.optimistic_gradient_descent.
#1400 opened Aug 6, 2025 by carlosgmartin Loading…
2
MuonClip draft in optax
#1383 opened Jul 15, 2025 by rdyro • Draft
4
Edit _schedule_free_test.py::ScheduleFreeTest::test_explicit_dtype to avoid DeprecationWarning about casting from complex to real dtypes.
#1380 opened Jul 11, 2025 by carlosgmartin Loading…
6
Fix dtype casting inside tree_add_scale.
#1376 opened Jul 11, 2025 by carlosgmartin Loading…
6
Fix tree_min and tree_max to handle zero-size array leafs.
#1374 opened Jul 6, 2025 by carlosgmartin Loading…
2
Deprecate optax.global_norm in favor of optax.tree.norm.
#1368 opened Jun 27, 2025 by carlosgmartin Loading…
2
add segmentation based (dice) loss
#1366 opened Jun 27, 2025 by aymuos15 Loading…
19
Minor improvements and cleanup using tree utility functions.
#1365 opened Jun 27, 2025 by carlosgmartin Loading…
2
Add more projection functions.
#1351 opened Jun 21, 2025 by carlosgmartin Loading…
8
Add tree_bits function.
#1341 opened Jun 18, 2025 by carlosgmartin Loading…
20
Silence some pytype errors related to a JAX build refactor
#1338 opened Jun 16, 2025 by copybara-service bot Loading…
Avoid unnecessary directory change in test.sh.
#1324 opened Jun 9, 2025 by carlosgmartin Loading…
Add weight_decay and mask arguments to adabelief optimizer.
#1291 opened May 1, 2025 by carlosgmartin Loading…
1
6
Add tests on python 3.13
#1272 opened Apr 18, 2025 by copybara-service bot Loading…
Add support for loss functions with auxiliary data to linesearch
#1177 opened Jan 16, 2025 by ro0mquy • Draft
4 tasks
10
Add linear program solver based on the restarted Halpern primal-dual hybrid gradient (rHPDHG) algorithm.
#1154 opened Dec 9, 2024 by carlosgmartin Loading…
1
Gauss-Newton and Levenberg-Marquardt
#920 opened Apr 9, 2024 by gbruno16 • Draft
Fix init_value and end_value in cosine decay
#884 opened Mar 26, 2024 by copybara-service bot Loading…
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
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…
15
Allow gradient transform parameters to be dynamic
#516 opened Mar 22, 2023 by NeilGirdhar Loading…
12
ProTip! Exclude everything labeled bug with -label:bug.