Thanks for the update.
I would still recommend to post the issue in the linked GitHub repo, if the error is specific to backpack.
If that’s not the case and you are seeing the issue using PyTorch only, could you post a code snippet to reproduce this issue, please?