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

Fix Muon implementation
#1170 opened Jan 9, 2025 by leloykun Loading…
removing scalar_type_of - random.key breaks it
#1169 opened Jan 9, 2025 by rdyro Loading…
replaced seed with key
#1167 opened Jan 6, 2025 by Tomas542 Loading…
Add non-negative least squares solver.
#1155 opened Dec 9, 2024 by carlosgmartin Loading…
Update references to JAX's GitHub repo
#1151 opened Dec 5, 2024 by copybara-service bot Loading…
Add improved version of Hungarian algorithm.
#1140 opened Nov 20, 2024 by carlosgmartin Loading…
[Triplet Margin Loss] Issue 1118
#1120 opened Oct 26, 2024 by cvnad1 Loading…
Adds AdeMAMix Optimizer to contrib
#1104 opened Oct 14, 2024 by mathDR Loading…
Simplification for transforms with empty state
#895 opened Mar 28, 2024 by fabianp Loading…
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…
Implementation of Eve Optimizer
#805 opened Feb 15, 2024 by amosyou Draft
Fix NamedTuples Formatting
#803 opened Feb 15, 2024 by amosyou Loading…
Remove TF dependency in reduce_on_plateau example
#768 opened Feb 4, 2024 by mmhamdy Loading…
ProTip! no:milestone will show everything without a milestone.