Search results

feat: adamw support for weight decay scheudles

Status: Open.

#1426 In google-deepmind/optax;

· Dhia-naouali opened on Oct 5, 2025

<code>value_and_grad_from_state</code> does not allow for specifying keyword arguments that <code>value_and_grad</code> uses

type:featureNew feature or requestNew feature or request

Status: Open.

#1418 In google-deepmind/optax;

· jna29 opened on Sep 6, 2025

Honor the dtype in the function signature in <code>inject_hyperparams</code>.

Status: Open.

#1414 In google-deepmind/optax;

· sizhit2 opened on Sep 3, 2025

Request: more permissible shape checking for regression losses (equal -&gt; broadcastable)

type:featureNew feature or requestNew feature or request

Status: Open.

#1408 In google-deepmind/optax;

· niamorg opened on Aug 22, 2025

DType of update depends on whether bfloat16 learning rate is passed as float, jax scalar or numpy scalar

type:others

Status: Open.

#1405 In google-deepmind/optax;

· colehaus opened on Aug 19, 2025

<code>piecewise_interpolate_schedule</code> sometimes generate nan learning rate if the number of transition steps is low.

type:bugSomething isn't workingSomething isn't working

Status: Open.

#1403 In google-deepmind/optax;

· VRichardJP opened on Aug 15, 2025

AMSGrad implementation differs from PyTorch/TensorFlow

type:featureNew feature or requestNew feature or request

Status: Open.

#1389 In google-deepmind/optax;

· sklenard opened on Jul 21, 2025

Publishing with <code>py.typed</code>

type:featureNew feature or requestNew feature or request

Status: Open.

#1388 In google-deepmind/optax;

· tmke8 opened on Jul 21, 2025

Add pre-commit hook that strips notebooks of metadata

dependenciesPull requests that update a dependency filePull requests that update a dependency file

Status: Open.

#1372 In google-deepmind/optax;

· carlosgmartin opened on Jul 6, 2025

Add segmentation based losses

type:featureNew feature or requestNew feature or request

Status: Open.

#1358 In google-deepmind/optax;

· aymuos15 opened on Jun 24, 2025

Minor suggestions for make_perturbed_fun

type:featureNew feature or requestNew feature or request

Status: Open.

#1342 In google-deepmind/optax;

· carlosgmartin opened on Jun 19, 2025

Feature request: Make documentation website have separate pages for each function

documentationImprovements or additions to documentationImprovements or additions to documentation

type:featureNew feature or requestNew feature or request

Status: Open.

#1320 In google-deepmind/optax;

· carlosgmartin opened on Jun 8, 2025