.. automodapi:: lib.torch_utils
   :include-all-objects:
   :no-inheritance-diagram: