IterModules

Recursively iterates over all nested Modules.

Abstract Signature:

IterModules(module: Module)

PyTorch

API: module.named_modules
Strategy: Macro '{module}.named_modules()'

Flax NNX

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