Model interpretability and understanding for PyTorch - View it on GitHub