Skip to content

Issues: google-deepmind/optax

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

37 Open 226 Closed
37 Open 226 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
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Bug in optax 0.2.4: Adam optimiser does not work in a jax tracer function, but Optax 0.2.3 does
#1159 opened Dec 19, 2024 by olive004
@rdyro
1
Add nnls (non-negative least squares)
#1152 opened Dec 8, 2024 by carlosgmartin
1
multitransform does not work with flax nnx
#1148 opened Dec 3, 2024 by MiladInk
1
When I install optax, I am no longer able to use the GPU
#1144 opened Nov 27, 2024 by Alessandro-Castelli
10
add_noise should take key as KeyArrayLike, not seed as int
#1137 opened Nov 13, 2024 by Tomas542
8
Possible bug: acprop explodes during initial steps
#1132 opened Nov 9, 2024 by carlosgmartin
Allow all optimizer update methods to receive an optional value argument
#1131 opened Nov 9, 2024 by carlosgmartin
3
Addition of NLLoss and its variant
#1124 opened Oct 29, 2024 by Saanidhyavats
2
[Feature] Triplet Marginal Loss
#1118 opened Oct 23, 2024 by cvnad1
2
Porting of OSQP solver
#1113 opened Oct 21, 2024 by xela-95
2
Updates dtype do not need to match params dtype but only grads dtype a priori
#1098 opened Oct 9, 2024 by vroulet
@vroulet
Each run of test.sh is very slow
#1091 opened Oct 5, 2024 by carlosgmartin
4
explain what optax.centralize does documentation Improvements or additions to documentation good first issue Good for newcomers
#1080 opened Oct 1, 2024 by fabianp
@miguelcsx
2
Feature request for the AdeMAMix optimizer.
#1058 opened Sep 13, 2024 by mathDR
2
4
Support for loss function with auxiliary data in linesearch
#1053 opened Sep 10, 2024 by ro0mquy
2
Feature request: GaLore optimizer
#1028 opened Jul 26, 2024 by gil2rok
4
Timeline for JaxOpt migration
#977 opened May 28, 2024 by Joshuaalbert
29
unitwise_norm fails for 3D convolutions enhancement New feature or request
#906 opened Apr 5, 2024 by froody
1
Inconsistencies in schedules API good first issue Good for newcomers
#835 opened Feb 29, 2024 by fabianp
3 tasks
5
Add a mathematical description of the algorithms documentation Improvements or additions to documentation good first issue Good for newcomers
#757 opened Feb 2, 2024 by vroulet
5 of 19 tasks
NamedTuples not being documented properly documentation Improvements or additions to documentation work-in-progress
#669 opened Dec 10, 2023 by fabianp
15
remove tensorflow as dependency for the examples
#656 opened Dec 5, 2023 by fabianp
@mmhamdy
7
New Example on creating a custom GradientTransformation documentation Improvements or additions to documentation
#652 opened Dec 2, 2023 by SauravMaheshkar
@SauravMaheshkar
2
DPSGD won't work correctly with apply_every
#527 opened Apr 23, 2023 by long21wt
MultiSteps doesn't work with DPSGD
#515 opened Mar 21, 2023 by long21wt
1
ProTip! Updated in the last three days: updated:>2024-12-21.