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
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
Add a `nesterov` flag to `radam` optimizer
#944
carlosgmartin
closed
2 months ago
2
Silence installs to improve readability of tests.
#943
copybara-service[bot]
closed
2 months ago
0
Fix pytype/pylint/deprecations warnings in tests.
#942
copybara-service[bot]
closed
2 months ago
0
Fix docs errors
#941
copybara-service[bot]
closed
1 month ago
0
fix(docs): broken link in README
#940
jeertmans
closed
2 months ago
0
chore(ci): add markdown-link-check action
#939
jeertmans
closed
2 months ago
0
Fix broken or slow notebook examples.
#938
copybara-service[bot]
closed
2 months ago
0
README: deadlink to "Optax 101" notebook
#937
jeertmans
closed
2 months ago
2
Add tree_l1_norm.
#936
copybara-service[bot]
closed
2 months ago
0
Internal change
#935
copybara-service[bot]
closed
1 month ago
0
Internal change
#934
copybara-service[bot]
closed
3 weeks ago
1
Make optax `jax.scipy.stats` lazy
#933
copybara-service[bot]
closed
2 months ago
0
Passing arguments to train multiple models in parallel
#932
kclauw
opened
2 months ago
4
Add a deprecation module to warn or raise errors for deprecations (following jax semantics).
#931
copybara-service[bot]
closed
2 months ago
0
Access subpackages directly from main module
#930
copybara-service[bot]
closed
2 months ago
0
Rename dpsgd to orbax.contrib END_PUBLIC
#929
copybara-service[bot]
closed
1 month ago
0
Rename DifferentiallyPrivateAggregateState to `orbax.contrib`
#928
copybara-service[bot]
closed
1 month ago
0
Rename differentially_private_aggregate -> optax.contrib
#927
copybara-service[bot]
closed
1 month ago
0
move clipping transforms to optax.transforms.
#926
copybara-service[bot]
closed
1 month ago
0
add tree_clip function.
#925
copybara-service[bot]
closed
2 months ago
0
Handling of -inf in cross entropy as a separate function.
#924
copybara-service[bot]
closed
2 months ago
0
Move gradient transformations to optax.transforms sub-package - 1/N
#923
copybara-service[bot]
closed
2 months ago
0
Fix pip install command for doc dependencies.
#922
mblondel
closed
2 months ago
0
Start documentation for projections.
#921
mblondel
closed
2 months ago
0
Gauss-Newton and Levenberg-Marquardt
#920
gbruno16
opened
2 months ago
0
Add projection_simplex.
#919
copybara-service[bot]
closed
2 months ago
0
Added mathematical documentation to AdaMax
#918
hmludwig
closed
2 months ago
1
Fix jax.tree_map deprecation warnings.
#917
copybara-service[bot]
closed
2 months ago
1
Correct handling of -inf in softmax_cross_entropy. Fix #898.
#916
copybara-service[bot]
closed
2 months ago
0
Revert of #898.
#915
copybara-service[bot]
closed
2 months ago
0
Reorganize tree_utils.
#914
copybara-service[bot]
closed
2 months ago
0
"masked" leads to unexpected behavior with callable Pytrees
#913
JadM133
opened
2 months ago
6
Add axis and where arguments to loss functions.
#912
carlosgmartin
opened
2 months ago
3
Port schedule_free optimizer to optax. Original pytorch repo: https://github.com/facebookresearch/schedule_free
#911
copybara-service[bot]
closed
2 months ago
10
Implement Schedule-Free Learning
#910
ameya98
closed
2 months ago
1
Update JAX dependency to >=0.4.26 so that we can access the jax.nn.sparse_plus function.
#909
copybara-service[bot]
closed
3 months ago
0
Suppress not-callable pylint error for now since is being flagged erroneously all over the place.
#908
copybara-service[bot]
closed
3 months ago
0
Implement 2 alternative conditional update wrappers.
#907
copybara-service[bot]
closed
2 months ago
0
unitwise_norm fails for 3D convolutions
#906
froody
opened
3 months ago
1
Documentation for cosine decay schedule
#905
gjhuizing
opened
3 months ago
2
Failure when running tests (due to new jax release?)
#904
fabian-sp
closed
3 months ago
2
Fix doc link
#903
yixiaoer
closed
3 months ago
0
Add `axis` and `where` arguments to loss functions
#902
carlosgmartin
opened
3 months ago
2
Fixed pseudocode for Nesterov in description of SGD.
#901
satyenkale
closed
2 months ago
1
Math pseudocode in description of SGD with Nesterov is incorrect
#900
satyenkale
closed
2 months ago
5
Upstream sparsemax jaxopt loss to optax.
#899
copybara-service[bot]
closed
2 months ago
0
Previous
Next