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