mlx.nn.Module.state

property Module.state

The module’s state dictionary

The module’s state dictionary contains any attribute set on the module including parameters in Module.parameters()

Unlike Module.parameters(), the Module.state property is a reference to the module’s state. Updates to it will be reflected in the original module.