jaxrts.experimental.SiiNN.NNModel.iter_modules

NNModel.iter_modules() → Iterator[tuple[tuple[Key, ...], Module]]

Warning: this method is method is deprecated; use iter_modules() instead.

Recursively iterates over all nested Module’s of the current Module, including the current Module. Alias of iter_modules().