Yeah it definitely seems like the same thing!
How do you approach do debug such problems? I would love to be a able to understand more about such issues by myself. As you might understand from my code, I tried graphing the computational graph using graphviz and debugging, but it wasn’t really helpful since the graph does not show which buffers are still in memory from my understanding. Any tips?