1 Pull request merged by 1 person

• Simplify ToNNX access of Linen module methods

  #4766 merged Jun 13, 2025

3 Pull requests opened by 2 people

• Fixes colab link for nnx docs

  #4775 opened Jun 10, 2025

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

  #4776 opened Jun 10, 2025

• Fix typo in Flax `nnx_basics` doc.

  #4781 opened Jun 15, 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

4 Issues opened by 4 people

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

  #4782 opened Jun 15, 2025

• nnx.scan working

  #4780 opened Jun 14, 2025

• Better type Module.apply signature - PR discussion

  #4779 opened Jun 13, 2025

• Deprecation of `nnx.Remat` and usage of `nnx.remat`

  #4778 opened Jun 11, 2025

5 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.

• Multi-host migration of `nnx.pmap` to `nnx.shard_map`

  #4751 commented on Jun 11, 2025 • 0 new comments

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

  #4774 commented on Jun 11, 2025 • 0 new comments

• Failed build: CI - with JAX nightly

  #4488 commented on Jun 16, 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 13, 2025 • 0 new comments