Collection of PyTorch-related utility functions - View it on GitHub