state

Contents

state#

class flax.nnx.State(mapping, /, *, _copy=True)[source]#

A pytree-like Mapping with hashable and comparable keys.