Hey @Jan_Balewski,
This might be related to this issue: [BUG] Probabilities do not sum to one with Torch · Issue #5444 · PennyLaneAI/pennylane · GitHub 
I’ll get back to you!
It seems I should start visiting Las Vegas - whatever I touch I hit a jackpot
Hi @Jan_Balewski,
A bugfix seems to be underway!
Keep an eye on the bugfix PR and when it’s merged you can try using the development version of PennyLane.
Please let us know in case you’re still experiencing issues after this.
Hi @Jan_Balewski, the fix has now been merged into the master branch of PennyLane!
Check out our install page (and click on Preview) if you don’t know how to use the development version of PennyLane.
And please let us know if your issue is now fixed or if it persists!