tensorkelechi

/

microdit_b

like 0

• "flax.nnx.statelib.State",
• "numpy.ndarray",
• "flax.nnx.rnglib.RngKey",
• "jax._src.prng.PRNGKeyArray",
• "numpy._core.multiarray._reconstruct",
• "flax.core.frozen_dict.FrozenDict",
• "jax._src.prng.threefry_split",
• "jax._src.prng.threefry_seed",
• "flax.nnx.variablelib.VariableState",
• "jax._src.prng.threefry_fold_in",
• "numpy.dtype",
• "jax._src.prng.threefry_random_bits",
• "flax.nnx.rnglib.RngCount",
• "flax.nnx.variablelib.Param",
• "jax._src.prng.PRNGImpl",
• "jax._src.array._reconstruct_array"

How to fix it?