Search results

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

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