pytorch-extras (pypi): A handful of extensions to the pytorch library.
Keywords: torch