PyTorch implementation of Keras already existing project: https://github.com/albermax/innvestigate/