PyTorch DAG Tracer -- Easy Visualization and Debugging

Hey everyone, I finished building a PyTorch Graph Tracer to make debugging easier! This tool visualizes the order in which tensors are created, making it simple to understand the flow and structure of your model. It’s a solid first version, and I’m excited to hear what you all think!

Feel free to test it out, share feedback or suggestions for improvement, and let me know if you find any bugs! I’d love to see how it can help with your PyTorch projects. :blush:

The code is in this link: 2manikan/Pytorch_DAG_Visualization_Tool

Note: For now, it works by installing PyTorch, cloning the repo, and keeping all the files in the same folder. The README has more details!