Improve error message for when nnx.Modules use jax or numpy arrays as leaf values#4492

Open

RaghuSpaceRajan wants to merge 6 commits intogoogle:mainfrom RaghuSpaceRajan:main

+2-1

Commits on Jan 20, 2025

Improve error message for when nnx.Modules use jax or numpy arrays as leaf values
RaghuSpaceRajan
authored

Commits on Jan 24, 2025