Skip to main content

ModelVision

Framework-agnostic neural network architecture visualizer. Renders publication-quality, fully styleable diagrams for models built in PyTorch, TensorFlow/Keras, JAX/Flax, JAX/Haiku, Hugging Face Transformers, scikit-learn, and ONNX — without running a forward pass.

Status: early development (Week 0 foundations landing; M1 next). See PRD_ModelVision.md for the full spec.

Install

uv add modelvision                    # core only
uv add "modelvision[torch]"           # per-framework extras
uv add "modelvision[all]"             # everything

Quick start

import torchvision.models as models
import modelvision as mvision

model = models.vgg16()
mvision.render(
    model,
    output="vgg16.svg",
    theme="dark",
    layer_palette={
        "Conv2d":    "#4a90d9",
        "ReLU":      "#27ae60",
        "MaxPool2d": "#e67e22",
        "Linear":    "#9b59b6",
    },
)

CLI

uvx modelvision model.py MyNet --output diagram.svg --theme dark
uvx modelvision model.onnx --output diagram.html

Development

git clone https://github.com/pianistprogrammer/ModelVision.git
cd modelvision
uv sync --all-extras --extra dev
uv run pytest

License

MIT

Metadata

Release files for modelvision 0.1.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for modelvision 0.1.0
File Size Uploaded Provenance Actions
modelvision-0.1.0.tar.gz 289.9 kB Without provenance Details for modelvision-0.1.0.tar.gz

Built distribution (wheel)

Table of built distributions (wheels) for modelvision 0.1.0
File Interpreter ABI Platform Actions
modelvision-0.1.0-py3-none-any.whl Python 3 none any Details for modelvision-0.1.0-py3-none-any.whl

Total release size: 381.5 kB

Release files / modelvision-0.1.0.tar.gz

Download URL modelvision-0.1.0.tar.gz
Size 289.9 kB
Tags Source
SHA-256 checksum
How to use checksums
d6d2a1e2e7fbf7d0e5c66a1cdd5632e70c84288759e841c1c776cc62365b3ac7
BLAKE2b-256 checksum
How to use checksums
19cf395a11bb3b692d02d9ad83447c0905b2d9483d66b20ce5269cd82cf06951
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via uv/0.11.20 {"installer":{"name":"uv","version":"0.11.20","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

Release files / modelvision-0.1.0-py3-none-any.whl

Download URL modelvision-0.1.0-py3-none-any.whl
Size 91.6 kB
Tags Python 3
SHA-256 checksum
How to use checksums
46171d1ecfb24f5e925e036533061ad5cea2d3504b1425726e0a3b6b3a77e668
BLAKE2b-256 checksum
How to use checksums
218bad8389bf082477e8b5ec2cb6d0541dd848512d00967a01d1c82ba4c6880f
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via uv/0.11.20 {"installer":{"name":"uv","version":"0.11.20","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

Security for modelvision 0.1.0

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Release history Release notifications | RSS feed

This release

0.1.0 This release

2 release files No provenance

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page