I get the intuition behind LRP 99 and I would like to implement in PyTorch. However, I’m not that familiar with the PyTorch internals to know how to best get started.
The website links to a LRP wrapper for Tensorflow. When looking at the code, it seems one has to wrap each layer to add some LRP specific code snippet. I assume the same would be required for PyTorch.
Does anyone already implemented LRP in Pytorch? Or does anyone has some opinions how to tackle this?