State

Extracts state dictionary from a node.

Abstract Signature:

State(node)

PyTorch

API: state_dict
Strategy: Plugin (torch_state_dict_to_nnx)

NumPy

API: numpy.random.bit_generator.SeedSequence.state
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.optimizers.optimizers.MultiOptimizer.state
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.state
Strategy: Direct Mapping
Official Docs ↗