IterGraph

Iterates over all nested nodes and leaves of the graph.

Abstract Signature:

IterGraph(node)

PyTorchFlax NNX

PyTorch

API: —

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

Flax NNX

API: flax.nnx.iter_graph

Strategy: Direct Mapping

Official Docs ↗