GitHub repository: Infinity-ops/captum
Primary language: Python
Description: Model interpretability and understanding for PyTorch