Search results

Inconsistent convergence behavior for <code>optax.optimistic_adam</code> on CPU and GPU

type:supportFurther information is requestedFurther information is requested

Status: Open.

#1387 In google-deepmind/optax;

· jingsenzhu opened on Jul 18, 2025

Lbfgs usage as Adam optimizer in flax.nnx

type:supportFurther information is requestedFurther information is requested

Status: Open.

#1386 In google-deepmind/optax;

· ak24watch opened on Jul 17, 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

Add optional weight decay to AdaBelief optimizer

type:featureNew feature or requestNew feature or request

Status: Open.

#1290 In google-deepmind/optax;

· carlosgmartin opened on May 1, 2025

Add missing projections from jaxopt

type:featureNew feature or requestNew feature or request

Status: Open.

#1280 In google-deepmind/optax;

· carlosgmartin opened on Apr 27, 2025

Muon for 4D parameters

type:featureNew feature or requestNew feature or request

Status: Open.

#1271 In google-deepmind/optax;

· ZagButNoZig opened on Apr 18, 2025

Rename <code>optax.perturbations</code> distribution method <code>log_prob</code> to <code>unnormalized_log_prob</code>

type:featureNew feature or requestNew feature or request

Status: Open.

#1214 In google-deepmind/optax;

· carlosgmartin opened on Mar 1, 2025

Allow for specifying axis for <code>optax.projections</code>

type:featureNew feature or requestNew feature or request

Status: Open.

#1182 In google-deepmind/optax;

· JesseFarebro opened on Jan 26, 2025

Addition of NLLoss and its variant

Status: Open.

#1124 In google-deepmind/optax;

· Saanidhyavats opened on Oct 29, 2024