exlib To create a dev environment: pip install '.[dev]' In order to use pytorch-gradcam, install the lab version to handle more diverse shapes at grad-cam@git+https://github.com/brachiolab/pytorch-grad-cam