pytorch\_probing package
========================

.. automodule:: pytorch_probing
   :members:
   :show-inheritance:
   :undoc-members:

.. toctree::
   :maxdepth: 4

   pytorch_probing.collect
   pytorch_probing.interceptor
   pytorch_probing.prober


.. toctree::
   :maxdepth: 4

   pytorch_probing.module_wrapper
   pytorch_probing.parallel_module_dict