model-sketch 0.0.1

A lightweight Python package that supplies batteries-included abstractions for PyTorch workflows

model-sketch

Image generated by AI

Python package for generating publication-quality visualizations of PyTorch model architectures. It focuses on clean SVG output, optional PNG export, and a simple API for sketching models from example forward inputs.

Installation

Base installation:

pip install model-sketch

If you also want PNG export support (CairoSVG backend), install the optional PNG extras:

pip install model-sketch[png]

Examples

Explore practical examples demonstrating model-sketch features:

Example	Description	Notebook	Colab
Tiny MLP sketch	Build a small MLP, render SVG/PNG, and save outputs to files.	simple_model.ipynb	Open in Colab