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
Extend example gallery entry for linear assignment problem with optimal transport example.
#1143
carlosgmartin
opened
4 hours ago
0
Fix complex support for L-BFGS
#1142
gautierronan
opened
1 day ago
2
L-BFGS optimizer with complex inputs
#1141
gautierronan
opened
2 days ago
2
Add improved version of Hungarian algorithm.
#1140
carlosgmartin
opened
3 days ago
3
`ocp.tree.serialize_tree` filtering logic for sequences with empty leaves
#1139
JesseFarebro
closed
5 days ago
2
Replaced seed with key in add_noise and noisy_sgd
#1138
Tomas542
opened
1 week ago
11
add_noise should take key as KeyArrayLike, not seed as int
#1137
Tomas542
opened
1 week ago
8
Beginning of 0.2.5 development
#1136
copybara-service[bot]
closed
1 week ago
0
Release v0.2.4
#1135
copybara-service[bot]
closed
1 week ago
0
Fix broken tests
#1134
copybara-service[bot]
closed
1 week ago
0
Revisiting linesearches and LBFGS.
#1133
copybara-service[bot]
closed
1 week ago
0
Possible bug: `acprop` explodes during initial steps
#1132
carlosgmartin
opened
2 weeks ago
0
Allow all optimizer `update` methods to receive an optional `value` argument
#1131
carlosgmartin
opened
2 weeks ago
3
Fix AdaBelief implementation.
#1130
carlosgmartin
closed
1 week ago
0
Format code base.
#1129
copybara-service[bot]
closed
2 weeks ago
0
Fix the sign for the update in the math equation for nadam in the docs.
#1128
carlosgmartin
closed
2 weeks ago
0
Add Nesterov momentum to AdaBelief optimizer.
#1127
carlosgmartin
closed
2 weeks ago
0
Add Muon Optimizer to `contrib`
#1126
leloykun
opened
3 weeks ago
4
Add warmup_constant_schedule to optax.__init__.py.
#1125
copybara-service[bot]
closed
3 weeks ago
1
Addition of NLLoss and its variant
#1124
Saanidhyavats
opened
3 weeks ago
2
remove test that leaked jax tracers
#1123
copybara-service[bot]
closed
3 weeks ago
0
fix: Coherent dtypes of updates with and without MultiSteps
#1122
hlzl
closed
2 weeks ago
0
Add a mathematical description for Lion
#1121
aman2304
closed
3 weeks ago
1
[Triplet Margin Loss] Issue 1118
#1120
cvnad1
opened
4 weeks ago
6
Support for Shampoo optimizer
#1119
kishorenc
closed
3 weeks ago
2
[Feature] Triplet Marginal Loss
#1118
cvnad1
opened
1 month ago
2
Add projection_linf_ball.
#1117
copybara-service[bot]
closed
1 month ago
0
Add tree_linf_norm.
#1116
copybara-service[bot]
closed
1 month ago
0
Add tree_max.
#1115
copybara-service[bot]
closed
1 month ago
0
Add projection_l2_sphere and projection_l2_ball.
#1114
copybara-service[bot]
closed
1 month ago
0
Porting of OSQP solver
#1113
xela-95
opened
1 month ago
2
Harmonize section order in docstrings
#1112
mblondel
closed
2 weeks ago
1
Rename pytree to tree in projection module
#1111
mblondel
closed
2 weeks ago
0
Changes to test.sh and pyproject.toml.
#1110
carlosgmartin
opened
1 month ago
12
multi_transform is inconsistent with a custom optimizer
#1109
idnm
closed
3 weeks ago
3
Avoid re-compilation and allow non-float32 dtypes in the zoom_linesearch
#1108
mirkobunse
closed
4 weeks ago
2
Fix doctest optimistic_adam
#1107
copybara-service[bot]
closed
1 month ago
0
Add projection_l1_sphere and projection_l1_ball.
#1106
copybara-service[bot]
closed
1 month ago
0
Fix typo in projection_simplex docstring.
#1105
copybara-service[bot]
closed
1 month ago
0
Adds AdeMAMix Optimizer to `contrib`
#1104
mathDR
opened
1 month ago
11
add config for link checking, and mark 429 (too many requests) as fine
#1103
fabianp
closed
1 month ago
0
Fix docstring for hungarian_algorithm.
#1102
carlosgmartin
closed
1 month ago
2
updates returns zeros
#1101
GeophyAI
closed
1 month ago
3
Add a projection onto L1 unit ball
#1100
jialinyi94
closed
1 month ago
2
Remove dtype safeguards
#1099
copybara-service[bot]
closed
1 month ago
0
Updates dtype do not need to match params dtype but only grads dtype a priori
#1098
vroulet
opened
1 month ago
0
cosmetic improvements perturbations module
#1097
fabianp
closed
1 month ago
0
use sphinx references instead of hardcoded links.
#1096
fabianp
closed
1 month ago
1
Fix formatting issues with gallery entry for linear assignment problem.
#1095
carlosgmartin
closed
1 month ago
0
update jax.tree.map to comply with jax 0.4.34
#1094
a1302z
closed
1 month ago
4
Next