EDM2's network doesn't support torch JVP due to usage of inplace "copy_" in MPConv #20
Description
Hello!
Thank you for your wonderful work and paper.
I am trying to use the EDM2 network for my own research project that utilizes a JVP.
However, due to torch's in-place "copy_" method being used in the MPConv layer, the JVP calculation fails.
Would it be possible to make changes to the MPConv layer such that it allows for JVP computation?
Thank you very much once again, and hope to hear a response :)
Best regards.