Skip to content

Pull requests: google-deepmind/optax

New pull request New
27 Open 668 Closed
27 Open 668 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

Fix error with x64 loss
#976 opened May 27, 2024 by stefanocortinovis Loading…
2
Add end_scale argument
#975 opened May 27, 2024 by stefanocortinovis Loading…
9
Fix memory leak in radam optimizer
#974 opened May 25, 2024 by lukekulik Loading…
4
Add ACProp optimizer.
#966 opened May 15, 2024 by carlosgmartin Loading…
15
feat: add scale_by_gradient_norm
#958 opened May 2, 2024 by SauravMaheshkar Loading…
3
Remove useless inner jit
#957 opened Apr 30, 2024 by copybara-service bot Loading…
ntxent fix
#946 opened Apr 23, 2024 by GrantMcConachie Loading…
6
Internal change
#935 opened Apr 17, 2024 by copybara-service bot Loading…
Internal change
#934 opened Apr 17, 2024 by copybara-service bot Loading…
Rename dpsgd to orbax.contrib END_PUBLIC
#929 opened Apr 12, 2024 by copybara-service bot Loading…
Rename DifferentiallyPrivateAggregateState to orbax.contrib
#928 opened Apr 12, 2024 by copybara-service bot Loading…
Rename differentially_private_aggregate -> optax.contrib
#927 opened Apr 12, 2024 by copybara-service bot Loading…
Gauss-Newton and Levenberg-Marquardt
#920 opened Apr 9, 2024 by gbruno16 • Draft
@vroulet
Add axis and where arguments to loss functions.
#912 opened Apr 8, 2024 by carlosgmartin Loading…
1
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
adding a perturbations module. Can take pytrees as inputs
#827 opened Feb 27, 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
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
Allow gradient transform parameters to be dynamic
#516 opened Mar 22, 2023 by NeilGirdhar Loading…
7
Adds Eve Optimizer needs work Needs more work by the author to be merged
#475 opened Jan 24, 2023 by wglao Loading…
4
ProTip! Adding no:label will show everything without a label.