Search results

Unnecessary check for trace level in nnx.Variable.__setattr__

Status: Open.

#4847 In google/flax;

· liudangyi opened on Jul 25, 2025

module with list of modules

Status: Open.

#4845 In google/flax;

· RisingPhoelix opened on Jul 24, 2025

<code>nnx.while_loop</code> does not work with <code>Scan over layers</code> or <code>nnx.remat</code>

Status: Open.

#4819 In google/flax;

· gliese581gg opened on Jul 16, 2025

Feature request, implement <code>nnx.PReLU</code>

Status: Open.

#4816 In google/flax;

· skutaada opened on Jul 15, 2025

optax lbfgs usage as adam , request for correct usage, as current version reports error

Status: Open.

#4812 In google/flax;

· ak24watch opened on Jul 10, 2025

Docstring for <code>flax.nnx.bridge.ToLinen</code> is incorrect

Status: Open.

#4803 In google/flax;

· jacknewsom opened on Jul 4, 2025

Typing.Protocol doesn't work. How to implement interfaces with nnx modules?

Status: Open.

#4789 In google/flax;

· DAOvb opened on Jun 24, 2025

Feature request, <code>nnx.Sequential()</code> should be the identity function.

Status: Open.

#4782 In google/flax;

· SobhanMP opened on Jun 15, 2025

nnx.scan working

Status: Open.

#4780 In google/flax;

· ak24watch opened on Jun 14, 2025

Better type Module.apply signature - PR discussion

Status: Open.

#4779 In google/flax;

· Daraan opened on Jun 13, 2025

Deprecation of <code>nnx.Remat</code> and usage of <code>nnx.remat</code>

Status: Open.

#4778 In google/flax;

· lraedel opened on Jun 11, 2025

nnx.jit() cannot specify backend or device due to out_shardings

Priority: P1 - soonResponse within 5 business days. Resolution within 30 days. (Assignee required)Response within 5 business days. Resolution within 30 days. (Assignee required)

Status: Open.

#4774 In google/flax;

· wookayin opened on Jun 9, 2025