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
165
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Upstream sparsemax jaxopt loss to optax.
#899
copybara-service[bot]
closed
3 months ago
0
Fix softmax_cross_entropy to handle -inf logits correctly when corresponding label is 0.
#898
carlosgmartin
closed
3 months ago
2
Added a NTXent loss
#897
GrantMcConachie
closed
2 months ago
6
softmax_cross_entropy handles -inf logits incorrectly
#896
carlosgmartin
closed
3 months ago
21
Simplification for transforms with empty state
#895
fabianp
opened
3 months ago
0
Add a mathematical description of AdamW
#894
gbruno16
closed
3 months ago
4
Beginning of 0.2.3 development
#893
copybara-service[bot]
closed
3 months ago
0
Release of version 0.2.2
#892
copybara-service[bot]
closed
3 months ago
0
Upstream multiclass_hinge_loss jaxopt loss to optax.
#891
copybara-service[bot]
closed
3 months ago
0
Upstream multiclass_perceptron_loss jaxopt loss to optax.
#890
copybara-service[bot]
closed
3 months ago
0
Document changes in power_iteration
#889
copybara-service[bot]
closed
3 months ago
0
Fix docs of tree_get, tree_set.
#888
copybara-service[bot]
closed
3 months ago
0
Fix notebook reduce_on_plateau
#887
copybara-service[bot]
closed
3 months ago
0
Fix the KeyboardInterrupt exception from https://github.com/google-deepmind/optax/issues/860 by removing the timeout
#886
copybara-service[bot]
closed
3 months ago
0
deprecate kwarg alpha in cosine_decay_schedule in favor of end_value
#885
copybara-service[bot]
closed
3 months ago
0
Fix init_value and end_value in cosine decay
#884
copybara-service[bot]
opened
3 months ago
0
Update reduce_on_plateau to handle training average loss.
#883
copybara-service[bot]
closed
3 months ago
0
Reccomendation for the `eps_root` setting for differentiating through the Adam optimizer
#882
itk22
closed
3 months ago
2
Restructure contrib folder
#880
copybara-service[bot]
closed
3 months ago
0
docstring cosmetic improvements
#879
fabianp
closed
3 months ago
0
Extend capabilities of tree_get, tree_set.
#878
copybara-service[bot]
closed
3 months ago
0
WIP chain optimizers in example reduce_on_plateau
#877
fabianp
closed
3 months ago
1
Docstring improvements to optax.chain
#876
copybara-service[bot]
closed
3 months ago
0
Fix error `reduce_on_plateau.ipynb:20002: WARNING: No source code lexer found for notebook cell`
#875
copybara-service[bot]
closed
3 months ago
0
Rename loss -> value in reduce_on_plateau
#874
copybara-service[bot]
closed
3 months ago
0
Expand the scope of tree_get and tree_set to handle subtrees and not only leaves.
#873
copybara-service[bot]
closed
3 months ago
0
Upstream missing jaxopt losses to optax - Part 2/N
#872
copybara-service[bot]
closed
3 months ago
0
Add filtering option to tree_get and tree_set.
#871
copybara-service[bot]
closed
3 months ago
0
deprecate kwarg alpha in cosine_decay_schedule in favor of end_value
#870
copybara-service[bot]
closed
3 months ago
0
Added mathematical description to AdaBelief docstring
#869
hmludwig
closed
3 months ago
0
Upstream missing jaxopt losses to optax.
#868
copybara-service[bot]
closed
3 months ago
0
FIX RST formatting in inject hyperparams
#867
fabianp
closed
3 months ago
0
[DOC] Add to the gallery an example on a small language model
#866
copybara-service[bot]
closed
3 months ago
0
Utility to set value in a pytree (and so in state)
#865
copybara-service[bot]
closed
3 months ago
0
Document the extra args of the update function in docstring
#864
copybara-service[bot]
closed
3 months ago
0
Warn that in future arguments after the initial (prediction, ground_truth) positional arguments will become keyword-only in optax losses.
#863
copybara-service[bot]
closed
3 months ago
0
Test on Python 3.12
#862
fabianp
opened
3 months ago
1
[IGNORE] dummy commit to see if RDT build upon pull requests
#861
fabianp
closed
3 months ago
0
Error in the lookahead optimizer example
#860
fabianp
closed
3 months ago
0
Utility to extract value from pytree (and so from state)
#859
copybara-service[bot]
closed
3 months ago
0
ENH: extend power_iteration to accept a matrix in implicit form
#858
copybara-service[bot]
closed
3 months ago
0
Added mathematical description to Noisy SGD
#857
hmludwig
closed
4 months ago
1
start of 0.2.2 development cycle
#856
copybara-service[bot]
closed
4 months ago
0
0.2.1 release
#855
copybara-service[bot]
closed
4 months ago
0
Broken tarball for release `v0.2.0`
#854
daskol
closed
4 months ago
3
Fix sign discrepancy in tolerance in scale_by_backtracking_linesearch
#853
copybara-service[bot]
closed
4 months ago
0
Document in docstring recent API changes in _tree_math.py
#852
copybara-service[bot]
closed
4 months ago
0
Reference reduce on plateau example from docstring
#851
copybara-service[bot]
closed
4 months ago
0
Break the scale_by_backtracking_linesearch docstring example for better readability
#850
copybara-service[bot]
closed
4 months ago
0
"dadapt_adamw, combine, prodigy" occur "TypeError: 'type' object is not subscriptable."
#849
sbb2002
closed
4 months ago
1
Previous
Next