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