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.1.tar.gz (14.9 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.1-py3-none-any.whl (13.1 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: metrik_torch-0.3.1.tar.gz
  • Upload date:
  • Size: 14.9 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.1.tar.gz
Algorithm Hash digest
SHA256 82b6d673aa90f270f3e2310858a84da1e9698064d5e5305220c33a8f4e96cdd4
MD5 61f7cac801101bdb82edde9615924724
BLAKE2b-256 f80faa47401ca14cb9f37ac2b552edf62e6aeaa5e9e7e27dae5e8617d406f172

See more details on using hashes here.

Provenance

The following attestation bundles were made for metrik_torch-0.3.1.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.1-py3-none-any.whl.

File metadata

  • Download URL: metrik_torch-0.3.1-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.1-py3-none-any.whl
Algorithm Hash digest
SHA256 04b860fec71d428da326bcfcd06ad823fe28566d76a06897f48e63a0fccb1970
MD5 6bb770a9004ad9c4953ecda0a142b5e1
BLAKE2b-256 ecf7dece108ee544aa71e2e9428836268a6c1932efc1ef5c0988ab721fd86edd

See more details on using hashes here.

Provenance

The following attestation bundles were made for metrik_torch-0.3.1-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