2 Pull requests merged by 2 people

• Use .input_formats and .output_formats in place of .input_layouts and .output_layouts respectively.

  #4784 merged Jun 19, 2025

• Simplify ToNNX access of Linen module methods

  #4766 merged Jun 13, 2025

4 Pull requests opened by 2 people

• Fix typo in Flax `nnx_basics` doc.

  #4781 opened Jun 15, 2025

• feat: Add overload signatures to module.apply

  #4783 opened Jun 18, 2025

• Use .format in place of .layout.

  #4785 opened Jun 19, 2025

• Internal changes

  #4786 opened Jun 20, 2025

1 Issue closed by 1 person

• Access to submodules/members of a linen module after using toNNX is not possible

  #4758 closed Jun 13, 2025

2 Issues opened by 2 people

• Feature request, `nnx.Sequential()` should be the identity function.

  #4782 opened Jun 15, 2025

• nnx.scan working

  #4780 opened Jun 14, 2025

6 Unresolved conversations

Sometimes conversations happen on old items that aren’t yet closed. Here is a list of all the Issues and Pull Requests with unresolved conversations.

• Problem training model with jax.jvp in forward pass

  #4636 commented on Jun 17, 2025 • 0 new comments

• Failed build: CI - with JAX nightly

  #4488 commented on Jun 20, 2025 • 0 new comments

• [nnx] mutable array p3

  #4755 commented on Jun 16, 2025 • 0 new comments

• Add training script for gemma example

  #4761 commented on Jun 17, 2025 • 0 new comments

• Fixes colab link for nnx docs

  #4775 commented on Jun 16, 2025 • 0 new comments

• Added a note to gemma guide about model's license consent on kaggle

  #4776 commented on Jun 16, 2025 • 0 new comments