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