Skip to main content

metrik-torch

Observed ModelGraph extraction from a real nn.Module tree.

$ pip install "metrik-torch[torch]"
$ metrik explore ./models/TinyLlama-1.1B --loader torch

The counterpart to metrik-explorer, which reads safetensors headers and config.json and never imports torch. This package walks the actual module tree, so module classes, parameter ownership, and weight tying are observed facts rather than claims from a config file.

Tying in particular: an untied lm_head has exactly the same shape as the embedding it does not share, so shape is not evidence. Here it is established by storage identity.

What it still does not know

Dataflow. nn.Module records containment, not what feeds what, so edges is None and fidelity is structural. Recovering real edges needs tracing, which breaks on the dynamic control flow most model implementations have. A consumer that needs edges checks fidelity and refuses rather than reading a containment tree as a dataflow graph.

No forward pass, no device transfer, no weight value read — module structure and parameter metadata only.

Why torch is an optional extra

uv sync --all-packages --dev stays torch-free so the CI matrix keeps running in about a minute. One dedicated job installs the extra and exercises extraction. See ADR 007.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

metrik_torch-0.3.0.tar.gz (14.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

metrik_torch-0.3.0-py3-none-any.whl (13.1 kB view details)

Uploaded Python 3

File details

Details for the file metrik_torch-0.3.0.tar.gz.

File metadata

  • Download URL: metrik_torch-0.3.0.tar.gz
  • Upload date:
  • Size: 14.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for metrik_torch-0.3.0.tar.gz
Algorithm Hash digest
SHA256 2415faf2ffef9e6c1f0001ba484468789a73405445bb94c2fba4307901f974f0
MD5 69a336324b6a5b5358424ce1de33520c
BLAKE2b-256 070162301a071a3cb5dc06d4524450494aece1df89e28b4d939ad1285937bfa3

See more details on using hashes here.

Provenance

The following attestation bundles were made for metrik_torch-0.3.0.tar.gz:

Publisher: release.yml on Asmodeus14/Metrik

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file metrik_torch-0.3.0-py3-none-any.whl.

File metadata

  • Download URL: metrik_torch-0.3.0-py3-none-any.whl
  • Upload date:
  • Size: 13.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for metrik_torch-0.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 2fb8a6dec69b958d72d88f70704142124072ffc38848fde28fd4dfbf90883bc7
MD5 be9bed7272fd7273145df8245f3c4db1
BLAKE2b-256 d09659009f18450e5674b36c553d87590343c82d4d89200a88fd065bd433df1e

See more details on using hashes here.

Provenance

The following attestation bundles were made for metrik_torch-0.3.0-py3-none-any.whl:

Publisher: release.yml on Asmodeus14/Metrik

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page