issues
search
google-deepmind
/
optax
Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.65k
stars
181
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Add Adan optimizer.
#1090
carlosgmartin
opened
19 hours ago
1
Add optax.optimistic_adam.
#1089
carlosgmartin
opened
21 hours ago
1
Fix initial step of scale_by_optimistic_gradient.
#1088
carlosgmartin
opened
1 day ago
1
Adopt US spelling for documentation and fix typos
#1087
miguelcsx
closed
1 day ago
0
Update docs: Note RMSprop usage instead of Adam for memory savings in…
#1086
nasyxx
closed
1 day ago
4
add link to examples from docstring
#1085
fabianp
closed
1 day ago
0
Adopt US spelling for the documentation
#1084
fabianp
closed
1 day ago
1
Add Hungarian algorithm for the linear assignment problem.
#1083
carlosgmartin
opened
2 days ago
2
Possible issue with the initial step of `scale_by_optimistic_gradient`
#1082
carlosgmartin
opened
2 days ago
2
Add Optimistic Adam
#1081
carlosgmartin
opened
2 days ago
3
explain what optax.centralize does
#1080
fabianp
opened
3 days ago
2
correct RST references
#1079
fabianp
closed
3 days ago
0
bump chex version for #1076
#1078
fabianp
closed
3 days ago
0
deprecate methods in `optax/monte_carlo`.
#1076
copybara-service[bot]
closed
2 days ago
0
Python 3.8 Compatibility Errors
#1075
jcraisbeck
closed
3 days ago
1
Tidy up test.sh and make it clean up properly.
#1074
carlosgmartin
closed
3 days ago
6
Fix doctests
#1073
copybara-service[bot]
closed
1 week ago
0
Fix test.sh to not modify .pylintrc.
#1071
carlosgmartin
closed
1 week ago
0
Fix test.sh, which uses set -u, so that it works when JAX_VERSION is unset.
#1070
carlosgmartin
closed
1 week ago
0
Relax absolute tolerance for failing tests involving chex.assert_trees_all_close.
#1069
carlosgmartin
closed
1 week ago
4
Replace deprecated typing.Hashable with collections.abc.Hashable.
#1068
carlosgmartin
closed
1 week ago
3
Running tests modifies .pylintrc and causes subsequent failures
#1067
carlosgmartin
closed
2 days ago
1
Migrate from jax.tree_util legacy APIs to new jax.tree API.
#1066
carlosgmartin
closed
1 week ago
0
Add missing initializer argument of 0 to tree_reduce in tree_vdot and tree_sum.
#1065
carlosgmartin
closed
3 days ago
4
Add optax.tree_utils.tree_random_split.
#1063
carlosgmartin
closed
1 week ago
1
Add utility to fetch common/lowest/highest dtype of a tree.
#1062
copybara-service[bot]
closed
1 week ago
0
LBFGS Linesearch
#1061
matillda123
closed
2 weeks ago
0
Ensure optimizers return updates of same dtype as params.
#1060
copybara-service[bot]
closed
1 week ago
0
schedule-free optimier: ensure it's possible to donate both the state and the params
#1059
enolan
closed
2 days ago
4
Feature request for the AdeMAMix optimizer.
#1058
mathDR
opened
3 weeks ago
0
Add AdEMAMix Optimizer
#1057
mathDR
closed
1 week ago
4
Add dtype option to tree_random_like
#1056
copybara-service[bot]
closed
2 weeks ago
0
Add double precision tests for safe_increment and fix warnings on float64_test.py
#1055
copybara-service[bot]
closed
2 weeks ago
0
Generalizes safe_int32_increment to safe_increment
#1054
copybara-service[bot]
closed
3 weeks ago
0
Support for loss function with auxiliary data in linesearch
#1053
ro0mquy
opened
3 weeks ago
2
misc improvements to optax's documentation
#1052
copybara-service[bot]
closed
3 weeks ago
0
Unwanted type promotion in prodigy optimizer
#1051
Grutschus
closed
1 week ago
1
more robust tests
#1050
copybara-service[bot]
closed
3 weeks ago
0
Modify clipping implementation to avoid jnp.moveaxis, which causes undesirable all-to-all's in distributed environments.
#1049
copybara-service[bot]
closed
2 weeks ago
1
Fix linting schedule_free_test
#1048
copybara-service[bot]
closed
4 weeks ago
0
Conversion to TFLite failed
#1047
JuanFMontesinos
closed
4 weeks ago
2
Params typed as any pytree.
#1046
copybara-service[bot]
opened
1 month ago
0
Update polynomial_schedule doctest per @vroulet's feedback
#1045
fabianp
closed
4 weeks ago
0
Fix typo in formula of cosine_decay_schedule
#1044
fabianp
closed
1 month ago
0
add missing schedule_free_* methods
#1043
fabianp
closed
1 month ago
0
schedule_free: fix broadcasting of scalar arrays to 1d arrays
#1042
n-gao
closed
1 month ago
11
fix error in softmax_cross_entropy formula
#1041
fabianp
closed
1 month ago
0
Wrong formula for `softmax_cross_entropy` in online doc
#1040
ziyuanzhao2000
closed
1 month ago
1
fix: MultiSteps with mixed precision params and updates.
#1039
hlzl
closed
1 week ago
14
Error when using `optax.Multisteps` with `optax.contrib.schedule_free`
#1038
ozanarmagan
closed
1 week ago
6
Next