ToPureDict

Converts a State object to a standard Python dictionary.

Abstract Signature:

ToPureDict(state)

PyTorchFlax NNX

PyTorch

API: state_dict

Strategy: Plugin (torch_state_dict_to_nnx)

Flax NNX

API: flax.nnx.to_pure_dict

Strategy: Direct Mapping

Official Docs ↗