for child_name, child in container.named_children():
child_fqn = f"{container_fqn}.{child_name}" if container_fqn else child_name
if is_inside_declared_subtree(child_fqn):
continue
blocks = list(child.named_children())
if not blocks:
continue
for block_name, block in blocks:
if id(block) in wrapped_module_ids:
continue
wrapped_module_ids.add(id(block))
wrapped_module = _get_checkpoint_wrapped_module(block)