Skip to content

Pull requests: google-deepmind/optax

New pull request New
48 Open 1,189 Closed
48 Open 1,189 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

Simplify freezing_tests
#1683 opened May 26, 2026 by copybara-service Bot Loading…
Fix pyrefly type-checking failures in GitHub CI.
#1682 opened May 26, 2026 by copybara-service Bot Loading…
Fix pytype and runtime errors in Optax causing GitHub CI failures.
#1681 opened May 26, 2026 by copybara-service Bot Loading…
Fix wrong schedule name in cosine_onecycle_schedule error message
#1680 opened May 24, 2026 by adityasingh2400 Loading…
1
docs: render one page per public symbol via autosummary toctree
#1679 opened May 24, 2026 by adityasingh2400 Loading…
1
4
Resolve various attribute access and overload warnings in losses, transforms, and contributed optimizers.
#1678 opened May 23, 2026 by copybara-service Bot Loading…
Resolve various type checking and indexing warnings in core optimizers and losses.
#1677 opened May 23, 2026 by copybara-service Bot Loading…
Repair use of deprecated parameter
#1671 opened May 5, 2026 by NeilGirdhar Contributor Loading…
6
feat(perturbations): rename sigma→scale, default num_samples=1 (#1342)
#1662 opened Apr 19, 2026 by yurekami Loading…
2 of 3 tasks
1
Issue #1658 - Add SSBroyden and SSBFGS optimizers to optax.contrib
#1659 opened Apr 10, 2026 by armbrusl Loading…
2
Add NorMuon optimizer (row-wise adaptive normalization for Muon)
#1651 opened Mar 30, 2026 by bzantium Contributor Loading…
4 tasks done
1
Remove deprecated second_order functions for 0.2.9
#1646 opened Mar 23, 2026 by dylantirandaz Loading…
2 tasks
1
Extend muon to support mask_callable weight dimension numbers
#1643 opened Mar 23, 2026 by dylantirandaz Loading…
3 tasks
4
Disable pmap-shard_map merge if it's available.
#1636 opened Mar 20, 2026 by copybara-service Bot Loading…
Standardize weight_decay_mask naming and add weight decay
#1635 opened Mar 20, 2026 by rdyro Collaborator Loading…
feat: add optional weight_decay to adabelief optimizer
#1634 opened Mar 19, 2026 by yasumorishima Loading…
3 tasks done
1
1
internal change
#1631 opened Mar 17, 2026 by copybara-service Bot Loading…
Adding pre-commit action to build markdown files derived from example jupyter notebooks.
#1627 opened Mar 12, 2026 by rdyro Collaborator Loading…
experimental notebook markdown preview generation via actions
#1626 opened Mar 12, 2026 by rdyro Collaborator Loading…
Enable performance gain by batching small update tensors.
#1618 opened Mar 5, 2026 by copybara-service Bot Loading…
1
Implement SOAP optimizer (#1580)
#1617 opened Mar 5, 2026 by jinukuntlaakhilakumargoud-web Loading…
1
Rename log_prob to unnormalized_log_prob in perturbations (#1214)
#1614 opened Mar 5, 2026 by jinukuntlaakhilakumargoud-web Loading…
1
3
Add PolarExpress variant to Muon optimizer
#1613 opened Mar 4, 2026 by MarcMachaczek Loading…
1
5
Add NorMuon optimizer
#1612 opened Feb 28, 2026 by tinker495 Contributor Loading…
3 tasks done
1
Add cautious weight decay transformation
#1611 opened Feb 28, 2026 by tinker495 Contributor Loading…
3 tasks done
@vroulet
2
ProTip! What’s not been updated in a month: updated:<2026-04-27.