_vf is a mapping to underlying cpp ATen library
I think in the end you’ll need to trace every operation inside RNN to find the origin of the problem
Meanwhile, you can replace your max(0, x) with torch.nn.functional.relu() but I doubt it will help