-
Notifications
You must be signed in to change notification settings - Fork 208
Pull requests: google-deepmind/optax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add example of reading learning rate from optimizer state
#363
opened Jun 18, 2022 by
atgctg
•
Review required
Adds Eve Optimizer
needs work
Needs more work by the author to be merged
#475
opened Jan 24, 2023 by
wglao
•
Review required
Allow gradient transform parameters to be dynamic
#516
opened Mar 22, 2023 by
NeilGirdhar
•
Review required
Hessian vector product and Gauss-Newton vector product utilities.
#817
opened Feb 21, 2024 by
copybara-service
bot
Loading…
Fix init_value and end_value in cosine decay
#884
opened Mar 26, 2024 by
copybara-service
bot
Loading…
Adding notebook illustrating schedule_free optimizer integration
#1022
opened Jul 24, 2024 by
copybara-service
bot
Loading…
Add linear program solver based on the restarted Halpern primal-dual hybrid gradient (rHPDHG) algorithm.
#1154
opened Dec 9, 2024 by
carlosgmartin
Loading…
Add alias for for functions in
optax.tree_utils
like jax.tree
#1166
opened Jan 3, 2025 by
ataa1312
Loading…
[#1196] implementing rename tree_scalar_mul to tree_scale
#1198
opened Feb 16, 2025 by
aryanmahawar205
Loading…
ProTip!
Mix and match filters to narrow down what you’re looking for.