Thanks @Shawn, that would be cool!
We have a guide here on the forum for contributing to PennyLane. However in this case, I think you’re right that it would be good to first get a handle on the errors that we get when trying to run TorchLayer on GPU and report them as an issue on the PennyLane GitHub. We can then decide from there whether there is a reasonable fix.