-
Notifications
You must be signed in to change notification settings - Fork 199
Pull requests: google-deepmind/optax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add alias for for functions in
optax.tree_utils
like jax.tree
#1166
opened Jan 3, 2025 by
Ata-Shaker
Loading…
Support tuple of axis in
softmax_cross_entropy_with_integer_labels
#1165
opened Jan 2, 2025 by
daskol
Loading…
Add batch_shape argument to optax.tree_utils.tree_random_like.
#1161
opened Dec 26, 2024 by
carlosgmartin
Loading…
Move configuration for Pylint from .pylintrc to pyproject.toml.
#1160
opened Dec 19, 2024 by
carlosgmartin
Loading…
Add linear program solver based on the restarted Halpern primal-dual hybrid gradient (rHPDHG) algorithm.
#1154
opened Dec 9, 2024 by
carlosgmartin
Loading…
Adding notebook illustrating schedule_free optimizer integration
#1022
opened Jul 24, 2024 by
copybara-service
bot
Loading…
Fix init_value and end_value in cosine decay
#884
opened Mar 26, 2024 by
copybara-service
bot
Loading…
Hessian vector product and Gauss-Newton vector product utilities.
#817
opened Feb 21, 2024 by
copybara-service
bot
Loading…
Previous Next
ProTip!
no:milestone will show everything without a milestone.