Show HN: TensorViz – Explore PyTorch models inside VS Code

Category: ai-ml

Tags: pytorch, visualization, vscode, machine-learning, devtools

Score: 5.3/10 (Innovation: 5, Technical: 6, Documentation: 3, Utility: 5)

TensorViz is a VS Code extension that lets developers explore PyTorch model architectures interactively, starting with GANs as a showcased example. It aims to bring model visualization directly into the IDE, which is a useful combo of devtools and ML workflows.

Target audience: machine learning engineers and data scientists

Repository: https://tensorviz.dev/explore/

View on Hacker News