Running
PyTorch Shape Lab
🧮
Follow a CNN's shapes without losing the batch dimension.
AI & ML interests
Tools for understanding and reviewing PyTorch models: nested modules, captured tensor shapes, source navigation and architecture changes inside VS Code.
Recent Activity
View all activity