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