pugh-torch (pypi): Functions, losses, and module blocks to share between experiments.
Keywords: pugh_torch