GitHub repository: sugatoray/captum
Description: Model interpretability and understanding for PyTorch