Thank you so much, that’s just what I needed!
Yeah, the nightly build version is cooler but I’ll stick with the standard one for now.
Side question: do you plan to add graph visualization tools directly to PyTorch?