state#
- class flax.nnx.State(mapping, /, *, _copy=True)[source]#
A pytree-like structure that contains a
Mappingfrom hashable and comparable keys to leaves. Leaves can be of any type butVariableStateandVariableare the most common.
A pytree-like structure that contains a Mapping from hashable and
comparable keys to leaves. Leaves can be of any type but VariableState
and Variable are the most common.