Add iter_module_children() to iterate over module children only.#5302

copybara-service[bot] wants to merge 1 commit intomaingoogle/flax:mainfrom

test_878177716google/flax:test_878177716Copy head branch name to clipboard