GetState

Retrieves the state (parameters/variables) from a model node.

Abstract Signature:

GetState(node, filters)

PyTorchKerasFlax NNX

PyTorch

API: —

Strategy: Macro '{node}.state_dict()'

Keras

API: —

Strategy: Macro '{node}.weights'

Flax NNX

API: flax.nnx.variables

Strategy: Direct Mapping

Official Docs ↗