fix: update JAX backend Module for flax nnx 0.12.0 strict Pytree compatibility#28933
Open
Sam-Armstrong wants to merge 1 commit into
Open
fix: update JAX backend Module for flax nnx 0.12.0 strict Pytree compatibility#28933Sam-Armstrong wants to merge 1 commit into
Sam-Armstrong wants to merge 1 commit into
