Kicel

/

economics3a

like 0

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

How to fix it?