Skip to content

Pull requests: google-deepmind/optax

New pull request New
28 Open 843 Closed
28 Open 843 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

Added mathematical description of NovoGrad optimizer
#1225 opened Mar 7, 2025 by shreyans413 Loading…
Renamed tree_scalar_mul to tree_scale and tree_add_scalar_mul to tree_add_scale with deprecation warnings (#1196)
#1223 opened Mar 6, 2025 by shreyans413 Loading…
1
type stability of state in optimizers, tested via jax.lax.cond
#1218 opened Mar 3, 2025 by copybara-service bot Loading…
Improve docstring for optax.power_iteration.
#1215 opened Mar 1, 2025 by carlosgmartin Loading…
6
Add more perturbation distributions to optax.perturbations.
#1213 opened Mar 1, 2025 by carlosgmartin Loading…
Add Simplified AdEMAMix.
#1206 opened Feb 24, 2025 by carlosgmartin Loading…
Update _combining.py changing examples to use optax.multi_transform, not optax.partition
#1205 opened Feb 23, 2025 by ywrt Loading…
6
Improve tests.sh.
#1202 opened Feb 19, 2025 by carlosgmartin Loading…
Fix nnls to handle batch shapes correctly.
#1200 opened Feb 17, 2025 by carlosgmartin Loading…
[#1196] implementing rename tree_scalar_mul to tree_scale
#1198 opened Feb 16, 2025 by aryanmahawar205 Loading…
1
1
Add support for loss functions with auxiliary data to linesearch
#1177 opened Jan 16, 2025 by ro0mquy • Draft
4 tasks
4
replaced seed with key
#1167 opened Jan 6, 2025 by Tomas542 Loading…
34
Add alias for for functions in optax.tree_utils like jax.tree
#1166 opened Jan 3, 2025 by ataa1312 Loading…
5
Add linear program solver based on the restarted Halpern primal-dual hybrid gradient (rHPDHG) algorithm.
#1154 opened Dec 9, 2024 by carlosgmartin Loading…
Update references to JAX's GitHub repo
#1151 opened Dec 5, 2024 by copybara-service bot Loading…
Adding notebook illustrating schedule_free optimizer integration
#1022 opened Jul 24, 2024 by copybara-service bot Loading…
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
ProTip! Type g i on any issue or pull request to go back to the issue listing page.