Search results

<code>nnx.PathContains</code> should support regex

Status: Open.

#5075 In google/flax;

· thijs-vanweezel opened on Nov 7, 2025

Migrate vae example from flax.linen to flax.nnx

Status: Open.

#5068 In google/flax;

· sanepunk opened on Nov 3, 2025

nnx.tabulate expects concrete values in unreleased flax&gt;0.12.0

Status: Open.

Bug

#5067 In google/flax;

· DBraun opened on Nov 3, 2025

Feature request: Semantic checkpoint in training loop

needs-discussion

Status: Open.

#5059 In google/flax;

· elly99-AI opened on Oct 29, 2025

Tensorflow test dependency needs upgrade to support orbax

Status: Open.

#5054 In google/flax;

· samanklesaria opened on Oct 27, 2025

nnx.remat (and other transformations) is easily misused.

Status: Open.

#5053 In google/flax;

· liudangyi opened on Oct 25, 2025

<code>nnx.scan</code> <code>sow</code> behavior changes when an <code>nnx.Pytree</code> is used as an explicit <code>nnx.Carry</code>

Status: Open.

Bug

#5051 In google/flax;

· maxencefaldor opened on Oct 24, 2025

Give RngStream the same interface as Jax's stateful RNGs

Status: Open.

#5046 In google/flax;

· samanklesaria opened on Oct 23, 2025

Provide a supported way to convert str keys from orbax checkpoints back to int keys

awaiting response

Status: Open.

#5030 In google/flax;

· pcmoritz opened on Oct 15, 2025

Python 3.14 compatibility issue

Status: Open.

Bug

#5027 In google/flax;

· burnpanck opened on Oct 14, 2025

How to construct freezable Parameters?

needs-discussion

Status: Open.

#4995 In google/flax;

· pfackeldey opened on Oct 3, 2025

Feature/Docs request: Saving and Loading checkpoints with optimizers and metrics

needs-discussion

Status: Open.

#4979 In google/flax;

· cdelv opened on Sep 24, 2025