probly.traverse_nn

Traverser utilities for neural networks.

Submodules

flax	Traversal implementation for flax modules.
reset_traverser	A wrapper around flextype.flexdispatch to create an extensible traverser with lazy type matching.
torch	Traversal implementation for PyTorch modules.
torch_utils	PyTorch-specific utils for traverse_nn.
utils	Generic utils for traverse_nn.