Search results

A big footgun in accumulation in adam like optimizers

type:supportFurther information is requestedFurther information is requested

Status: Open.

#1552 In google-deepmind/optax;

· AakashKumarNain opened on Jan 6, 2026

Feature Request: add MixMatch and FixMatch training support for semi-supervised learning (SSL)

type:featureNew feature or requestNew feature or request

Status: Open.

#1550 In google-deepmind/optax;

· surajyadav-research opened on Jan 6, 2026

Feature request: add BYOL, SimSiam, DINO and Barlow Twins losses to <code>optax.losses</code>

type:featureNew feature or requestNew feature or request

Status: Open.

#1528 In google-deepmind/optax;

· surajyadav-research opened on Dec 10, 2025

Code for DoG is duplicated

type:others

Status: Open.

#1509 In google-deepmind/optax;

· carlosgmartin opened on Nov 23, 2025

Should <code>tree_sum</code> use <code>jax.tree.reduce_associative</code>?

type:supportFurther information is requestedFurther information is requested

Status: Open.

#1498 In google-deepmind/optax;

· SobhanMP opened on Nov 17, 2025

Add Implementation of _Lion Optimizer (Evolved sign Momentum)

type:featureNew feature or requestNew feature or request

Status: Open.

#1437 In google-deepmind/optax;

· raghulchandramouli opened on Oct 24, 2025

Correct usage of look-ahead optimizer

type:supportFurther information is requestedFurther information is requested

Status: Open.

#1429 In google-deepmind/optax;

· SNMS95 opened on Oct 7, 2025

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

type:featureNew feature or requestNew feature or request

Status: Open.

#1414 In google-deepmind/optax;

· sizhit2 opened on Sep 3, 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