issues
search
google-deepmind
/
optax
Optax is a gradient processing and optimization library for JAX.
https://optax.readthedocs.io
Apache License 2.0
1.56k
stars
166
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Loose dependency allows chex version without 'warn_deprecated_function' to be installed
#995
phcavelar
closed
4 days ago
1
Support for CSR format sparse matrix in optimizer?
#994
MoFHeka
opened
5 days ago
1
Performance issue: multi_transform and set_to_zero don't prevent computation
#993
YanisJouanaud
opened
6 days ago
0
Add signSGD optimizer
#992
wtong98
opened
1 week ago
1
LBFGS part 4: notebook illustrating how to use lbfgs with linesearch as a solver.
#991
copybara-service[bot]
opened
1 week ago
0
AttributeError: module 'optax' has no attribute 'lbfgs'
#990
pascal-mueller
closed
6 days ago
2
LBFGS part 3: combine lbfgs and zoom linesearch
#989
copybara-service[bot]
closed
1 week ago
0
Internal change
#988
copybara-service[bot]
closed
2 weeks ago
0
CI: add test against oldest supported JAX version
#987
jakevdp
closed
2 weeks ago
1
Ignore some linesearch tests on gpu/tpu
#986
copybara-service[bot]
closed
2 weeks ago
0
Fix typing_extensions error
#985
copybara-service[bot]
closed
3 weeks ago
0
Fix docs errors (following warnings displayed in doc logs of github actions)
#984
copybara-service[bot]
closed
3 weeks ago
0
[JAX] Update users of jax.tree.map() to be more careful about how they handle Nones.
#983
copybara-service[bot]
closed
3 weeks ago
0
Fix LaTeX rendering issue in docstring of lbfgs in aliases for optimi…
#982
mcharrak
closed
5 days ago
3
LBFGS solver part 2: implementing linesearch ensuring sufficient decrease and small curvature
#981
copybara-service[bot]
closed
2 weeks ago
0
LBFGS solver part 1: chainable preconditioner.
#980
copybara-service[bot]
closed
3 weeks ago
0
Add sophia-h optimizer
#979
evanatyourservice
opened
1 month ago
6
Expose components in sub-package
#978
copybara-service[bot]
closed
1 month ago
0
Timeline for JaxOpt migration
#977
Joshuaalbert
opened
1 month ago
8
Fix error with x64 loss
#976
stefanocortinovis
closed
4 weeks ago
2
Add end_scale argument
#975
stefanocortinovis
closed
4 weeks ago
9
Fix memory leak in radam optimizer
#974
lukekulik
closed
1 month ago
3
Removing sophia optimizer
#973
copybara-service[bot]
closed
1 month ago
0
Fix broken test
#972
copybara-service[bot]
closed
1 month ago
0
Add multiclass_sparsemax_loss.
#971
copybara-service[bot]
closed
1 month ago
0
Fix typos in classification loss docstrings.
#970
copybara-service[bot]
closed
1 month ago
0
Memory leak in optax.radam
#969
lukekulik
closed
1 month ago
2
Intended usage of the Sophia optimiser
#968
vvvm23
opened
1 month ago
7
remove residues from previous builds before running tests
#967
fabianp
closed
1 month ago
0
Add ACProp optimizer.
#966
carlosgmartin
closed
3 weeks ago
12
Add ACProp
#965
carlosgmartin
closed
1 month ago
1
How to Cleanly Specify Optimizer, Schedule, & Gradient Clipping with inject_hyperparams
#964
timothyas
closed
1 month ago
3
Replace deprecated `jax.tree_*` functions with `jax.tree.*`
#963
copybara-service[bot]
closed
1 month ago
0
RST formatting in mechanic
#962
copybara-service[bot]
closed
1 month ago
0
LinearSolveTest.test_solve_sparse fails with jax 0.4.26
#960
GaetanLepage
closed
1 month ago
1
Jaxopt vs Optax
#959
pushkar5586
closed
1 month ago
3
feat: add `normalize_by_update_norm`
#958
SauravMaheshkar
opened
2 months ago
11
Remove useless inner jit
#957
copybara-service[bot]
closed
1 month ago
0
remove duplicated BATCH_SIZE argument
#956
fabianp
closed
2 months ago
0
Add safeguard warmup option in prodigy.
#955
copybara-service[bot]
closed
2 months ago
3
Add an assignment problem solver
#954
carlosgmartin
opened
2 months ago
0
Fix RST formatting issues.
#953
fabianp
closed
2 months ago
0
docstring formatting
#952
fabianp
closed
2 months ago
0
Adding DoG and DoWG.
#951
copybara-service[bot]
closed
2 months ago
0
Formatting in momo docstring + doctest
#950
fabianp
closed
2 months ago
0
Add a nesterov flag to radam optimizer.
#949
carlosgmartin
closed
2 months ago
0
Let momo accept additional extra arguments.
#948
copybara-service[bot]
closed
2 months ago
0
Add schedule possibility to cocob
#947
copybara-service[bot]
closed
2 months ago
0
ntxent fix
#946
GrantMcConachie
opened
2 months ago
8
Weight decay for COCOB
#945
albcab
closed
2 months ago
1
Next