probly.traverse_nn

Traverser utilities for neural networks.

Modules

common	Generic traverser helpers for neural networks.
flax	Traversal implementation for flax modules.
torch	Traversal implementation for PyTorch modules.