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