IterChildren

Iterates over immediate children modules/layers.

Abstract Signature:

IterChildren(module: Module)

PyTorch

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

Flax NNX

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