LICENSE
MANIFEST.in
NOTICE
README.md
pyproject.toml
torchlens/__init__.py
torchlens/__main__.py
torchlens/_architecture.py
torchlens/_capture_fingerprint.py
torchlens/_capture_honesty.py
torchlens/_capture_intervention.py
torchlens/_capture_state_helpers.py
torchlens/_chunked_capture_helpers.py
torchlens/_chunking.py
torchlens/_deploy_env.py
torchlens/_deprecations.py
torchlens/_distributed.py
torchlens/_episode_spec.py
torchlens/_errors.py
torchlens/_extraction_provenance.py
torchlens/_fast_run.py
torchlens/_input_coerce.py
torchlens/_input_walk.py
torchlens/_literals.py
torchlens/_model_wrappers.py
torchlens/_option_receipt.py
torchlens/_options_validation.py
torchlens/_plain_attr_fidelity.py
torchlens/_robustness.py
torchlens/_runnable_attestation.py
torchlens/_runnable_call_arguments.py
torchlens/_runnable_call_outputs.py
torchlens/_runnable_execution.py
torchlens/_runnable_input_aliases.py
torchlens/_runnable_input_metadata.py
torchlens/_runnable_input_sites.py
torchlens/_runnable_output_contracts.py
torchlens/_runnable_path_faithfulness.py
torchlens/_runnable_providers.py
torchlens/_runnable_seam.py
torchlens/_runnable_state.py
torchlens/_runnable_state_context.py
torchlens/_runnable_transaction.py
torchlens/_runnable_verification.py
torchlens/_runnable_witness_contracts.py
torchlens/_save_budget.py
torchlens/_selection_align.py
torchlens/_source_links.py
torchlens/_split_rebind.py
torchlens/_state.py
torchlens/_surface.py
torchlens/_trace_selector_helpers.py
torchlens/_trace_state.py
torchlens/_training_validation.py
torchlens/_transport.py
torchlens/_user_public_impls.py
torchlens/brainpipe.py
torchlens/captured_run.py
torchlens/constants.py
torchlens/dataset_extraction.py
torchlens/differential.py
torchlens/facets.py
torchlens/features.py
torchlens/hash.py
torchlens/inventory.py
torchlens/kernel_telemetry.py
torchlens/observers.py
torchlens/options.py
torchlens/py.typed
torchlens/quantities.py
torchlens/runnable.py
torchlens/selection.py
torchlens/selection_compare.py
torchlens/selection_graph.py
torchlens/selection_subspace.py
torchlens/selection_values.py
torchlens/trace_slice.py
torchlens/types.py
torchlens/user_funcs.py
torchlens.egg-info/PKG-INFO
torchlens.egg-info/SOURCES.txt
torchlens.egg-info/dependency_links.txt
torchlens.egg-info/entry_points.txt
torchlens.egg-info/requires.txt
torchlens.egg-info/top_level.txt
torchlens/_data_substrate/__init__.py
torchlens/_data_substrate/artifact.py
torchlens/_data_substrate/digests.py
torchlens/_data_substrate/model_identity.py
torchlens/_extraction/__init__.py
torchlens/_extraction/callable_identity.py
torchlens/_extraction/context.py
torchlens/_extraction/dtype_policy.py
torchlens/_extraction/engine.py
torchlens/_extraction/envelope.py
torchlens/_extraction/export.py
torchlens/_extraction/harvest_schema.py
torchlens/_extraction/pool.py
torchlens/_extraction/ragged.py
torchlens/_extraction/reader.py
torchlens/_extraction/resume.py
torchlens/_extraction/selector_plan.py
torchlens/_extraction/shards.py
torchlens/_extraction/views.py
torchlens/_io/__init__.py
torchlens/_io/_artifact_anchors.py
torchlens/_io/_artifact_strings.py
torchlens/_io/_canonical_pickle.py
torchlens/_io/_durability.py
torchlens/_io/_forgery_identity_facts.py
torchlens/_io/_inline_tensor.py
torchlens/_io/_json.py
torchlens/_io/_load_validators.py
torchlens/_io/_portability_preflight.py
torchlens/_io/_safe_unpickle.py
torchlens/_io/_save_gates.py
torchlens/_io/_source_privacy.py
torchlens/_io/_torch_symbols.py
torchlens/_io/accessor_rebuild.py
torchlens/_io/bundle.py
torchlens/_io/bundle_metadata.py
torchlens/_io/compat_ledger.py
torchlens/_io/field_registry.py
torchlens/_io/forgery_validation.py
torchlens/_io/format_contract.py
torchlens/_io/format_errors.py
torchlens/_io/injection_codec.py
torchlens/_io/lazy.py
torchlens/_io/manifest.py
torchlens/_io/paths.py
torchlens/_io/payload_codec.py
torchlens/_io/payload_reader.py
torchlens/_io/prerelease.py
torchlens/_io/rehydrate.py
torchlens/_io/runnable.py
torchlens/_io/runnable_coherence.py
torchlens/_io/runnable_load.py
torchlens/_io/save_policy.py
torchlens/_io/scrub.py
torchlens/_io/sidecar.py
torchlens/_io/state_contract.py
torchlens/_io/state_keys.py
torchlens/_io/streaming.py
torchlens/_io/streaming_async.py
torchlens/_io/tensor_policy.py
torchlens/_io/tlspec.py
torchlens/_io/writer_contract.py
torchlens/_registry/__init__.py
torchlens/_registry/kernel.py
torchlens/_trace_core/__init__.py
torchlens/_trace_core/columns.py
torchlens/_trace_core/core.py
torchlens/_trace_core/fact_blocks.py
torchlens/_trace_core/groups.py
torchlens/_trace_core/op_store.py
torchlens/_trace_core/overlays.py
torchlens/_trace_core/payloads.py
torchlens/_trace_core/pools.py
torchlens/_trace_core/record_rows.py
torchlens/_trace_core/relation_views.py
torchlens/_trace_core/relations.py
torchlens/_vocab/__init__.py
torchlens/_vocab/node_spec.py
torchlens/_vocab/trace_state.py
torchlens/accessors/__init__.py
torchlens/agent/__init__.py
torchlens/agent/_apimap.py
torchlens/agent/_artifacts.py
torchlens/agent/_budgets.py
torchlens/agent/_compare.py
torchlens/agent/_envelope.py
torchlens/agent/_fold.py
torchlens/agent/_guide_text.py
torchlens/agent/_overview.py
torchlens/agent/_query.py
torchlens/agent/_registry.py
torchlens/agent/_schemas.py
torchlens/agent/_stats.py
torchlens/agent/cli.py
torchlens/attribution/__init__.py
torchlens/attribution/_binder.py
torchlens/attribution/_core.py
torchlens/attribution/_gradient_shap.py
torchlens/attribution/_guided.py
torchlens/attribution/_layer.py
torchlens/attribution/_metrics.py
torchlens/attribution/_noise_tunnel.py
torchlens/attribution/_occlusion.py
torchlens/attribution/_result.py
torchlens/attribution/_sampling.py
torchlens/attribution/_stash.py
torchlens/attribution/_steps.py
torchlens/attribution/_text.py
torchlens/attribution/_viz.py
torchlens/attribution/onebackward/__init__.py
torchlens/attribution/onebackward/_accessor.py
torchlens/attribution/onebackward/_edge_plumbing.py
torchlens/attribution/onebackward/_engine.py
torchlens/attribution/onebackward/_errors.py
torchlens/attribution/onebackward/_facet_bridge.py
torchlens/attribution/onebackward/_frozen.py
torchlens/attribution/onebackward/_read.py
torchlens/attribution/onebackward/_suppress.py
torchlens/attribution/onebackward/_table.py
torchlens/attribution/onebackward/_targets.py
torchlens/autoroute/__init__.py
torchlens/autoroute/_builtin_input.py
torchlens/autoroute/_builtin_output.py
torchlens/autoroute/_registry.py
torchlens/autoroute/data/README.md
torchlens/autoroute/data/imagenet1k_labels.json
torchlens/autoroute/input/__init__.py
torchlens/autoroute/output/__init__.py
torchlens/backends/__init__.py
torchlens/backends/_finalize.py
torchlens/backends/_options.py
torchlens/backends/_protocol.py
torchlens/backends/_validation_shared.py
torchlens/backends/default_specs.py
torchlens/backends/registry.py
torchlens/backends/jax/__init__.py
torchlens/backends/jax/_site_dialect.py
torchlens/backends/jax/backend.py
torchlens/backends/jax/capabilities.py
torchlens/backends/jax/jaxpr.py
torchlens/backends/jax/modules.py
torchlens/backends/mlx/__init__.py
torchlens/backends/mlx/backend.py
torchlens/backends/mlx/capabilities.py
torchlens/backends/mlx/interventions.py
torchlens/backends/mlx/model_prep.py
torchlens/backends/mlx/tensor_store.py
torchlens/backends/mlx/validation.py
torchlens/backends/mlx/wrappers.py
torchlens/backends/paddle/__init__.py
torchlens/backends/paddle/_param_writes.py
torchlens/backends/paddle/backend.py
torchlens/backends/paddle/capabilities.py
torchlens/backends/paddle/interventions.py
torchlens/backends/paddle/model_prep.py
torchlens/backends/paddle/tensor_store.py
torchlens/backends/paddle/validation.py
torchlens/backends/paddle/wrappers.py
torchlens/backends/tf/__init__.py
torchlens/backends/tf/_tf_compat.py
torchlens/backends/tf/backend.py
torchlens/backends/tf/capabilities.py
torchlens/backends/tf/derived_grads.py
torchlens/backends/tf/funcgraph.py
torchlens/backends/tf/interventions.py
torchlens/backends/tf/modules.py
torchlens/backends/tf/op_callback_capture.py
torchlens/backends/tf/validation.py
torchlens/backends/tinygrad/__init__.py
torchlens/backends/tinygrad/backend.py
torchlens/backends/tinygrad/capabilities.py
torchlens/backends/torch/__init__.py
torchlens/backends/torch/_aten_capture.py
torchlens/backends/torch/_autograd_grad_boundary.py
torchlens/backends/torch/_completeness_audited.py
torchlens/backends/torch/_completeness_boundaries.py
torchlens/backends/torch/_completeness_cross_thread.py
torchlens/backends/torch/_completeness_dispatch.py
torchlens/backends/torch/_completeness_dispatch_names.py
torchlens/backends/torch/_completeness_escape_state.py
torchlens/backends/torch/_completeness_finalize.py
torchlens/backends/torch/_completeness_metadata.py
torchlens/backends/torch/_completeness_origins.py
torchlens/backends/torch/_completeness_patches.py
torchlens/backends/torch/_completeness_storage.py
torchlens/backends/torch/_completeness_types.py
torchlens/backends/torch/_fire_timing.py
torchlens/backends/torch/_function_boundary.py
torchlens/backends/torch/_gradfn_markers.py
torchlens/backends/torch/_held_plain_holders.py
torchlens/backends/torch/_held_refs.py
torchlens/backends/torch/_held_refs_capture.py
torchlens/backends/torch/_modes.py
torchlens/backends/torch/_module_arg_stubs.py
torchlens/backends/torch/_module_boundary_adoption.py
torchlens/backends/torch/_op_markers.py
torchlens/backends/torch/_ops_activations.py
torchlens/backends/torch/_ops_arguments.py
torchlens/backends/torch/_ops_autograd.py
torchlens/backends/torch/_ops_capture_records.py
torchlens/backends/torch/_ops_container_base.py
torchlens/backends/torch/_ops_containers.py
torchlens/backends/torch/_ops_emission.py
torchlens/backends/torch/_ops_exhaustive.py
torchlens/backends/torch/_ops_finalize.py
torchlens/backends/torch/_ops_interventions.py
torchlens/backends/torch/_ops_predicate_events.py
torchlens/backends/torch/_ops_predicates.py
torchlens/backends/torch/_ops_retention.py
torchlens/backends/torch/_ops_shared_fields.py
torchlens/backends/torch/_predicate_boundary_replacements.py
torchlens/backends/torch/_session_forwards.py
torchlens/backends/torch/_tl.py
torchlens/backends/torch/_torch_ops_calls.py
torchlens/backends/torch/_view_writeback.py
torchlens/backends/torch/_weightsfree_ctx.py
torchlens/backends/torch/_weightsfree_transparency.py
torchlens/backends/torch/aliasing.py
torchlens/backends/torch/backend.py
torchlens/backends/torch/backward.py
torchlens/backends/torch/belt.py
torchlens/backends/torch/bound_root.py
torchlens/backends/torch/buffer_writes.py
torchlens/backends/torch/collectives.py
torchlens/backends/torch/completeness_witness.py
torchlens/backends/torch/escape_detection.py
torchlens/backends/torch/funcol.py
torchlens/backends/torch/identity_shims.py
torchlens/backends/torch/legacy_ctors.py
torchlens/backends/torch/model_prep.py
torchlens/backends/torch/module_stack.py
torchlens/backends/torch/offload_hooks.py
torchlens/backends/torch/ops.py
torchlens/backends/torch/param_mutation.py
torchlens/backends/torch/prehook_provenance.py
torchlens/backends/torch/rescue.py
torchlens/backends/torch/sources.py
torchlens/backends/torch/structure_only_belt.py
torchlens/backends/torch/tensor_tracking.py
torchlens/backends/torch/wrappers.py
torchlens/bridge/__init__.py
torchlens/bridge/_contrastive.py
torchlens/bridge/_utils.py
torchlens/bridge/brain_score.py
torchlens/bridge/captum.py
torchlens/bridge/depyf.py
torchlens/bridge/dialz.py
torchlens/bridge/gradcam.py
torchlens/bridge/hf.py
torchlens/bridge/huggingface.py
torchlens/bridge/inseq.py
torchlens/bridge/mcp.py
torchlens/bridge/nnsight.py
torchlens/bridge/profiler.py
torchlens/bridge/repeng.py
torchlens/bridge/rsatoolbox.py
torchlens/bridge/sae.py
torchlens/bridge/sae_lens.py
torchlens/bridge/shap.py
torchlens/bridge/steering_vectors.py
torchlens/bridge/treescope.py
torchlens/bridge/xarray.py
torchlens/bridge/lit/__init__.py
torchlens/bridge/lit/_adapters.py
torchlens/bridge/lit/_pooling.py
torchlens/bridge/lit/_refusals.py
torchlens/bridge/lit/_runtime.py
torchlens/bridge/lit/_sites.py
torchlens/bundle/__init__.py
torchlens/bundle/_bytes.py
torchlens/bundle/_compare_gate.py
torchlens/bundle/_comparisons.py
torchlens/bundle/_deltas.py
torchlens/bundle/_lineage.py
torchlens/bundle/_outcome_fold.py
torchlens/bundle/_provenance.py
torchlens/bundle/_relation_carriage.py
torchlens/bundle/_relations.py
torchlens/bundle/_vary.py
torchlens/callbacks/__init__.py
torchlens/callbacks/lightning.py
torchlens/capture/__init__.py
torchlens/capture/_annotations_travel.py
torchlens/capture/_episode_coupling.py
torchlens/capture/_episode_derivation.py
torchlens/capture/_episode_failed.py
torchlens/capture/_episode_fold.py
torchlens/capture/_episode_join.py
torchlens/capture/_episode_ledger.py
torchlens/capture/_episode_ledger_anchors.py
torchlens/capture/_nonfinite_prefix.py
torchlens/capture/_plan_check.py
torchlens/capture/_structure_evidence.py
torchlens/capture/_structure_only_bridge.py
torchlens/capture/_structure_only_entry.py
torchlens/capture/_weightsfree_admission.py
torchlens/capture/arg_positions.py
torchlens/capture/compute_record.py
torchlens/capture/config.py
torchlens/capture/flops.py
torchlens/capture/outcome.py
torchlens/capture/peak_memory.py
torchlens/capture/plan.py
torchlens/capture/predicates.py
torchlens/capture/preflight.py
torchlens/capture/projections.py
torchlens/capture/projectors.py
torchlens/capture/salient_args.py
torchlens/capture/session.py
torchlens/capture/stop.py
torchlens/capture/structure_only.py
torchlens/capture/trace.py
torchlens/checks/__init__.py
torchlens/checks/_adapters.py
torchlens/checks/_audit.py
torchlens/checks/_constants.py
torchlens/checks/_errors.py
torchlens/checks/_ledgers.py
torchlens/checks/_param_checks.py
torchlens/checks/_range.py
torchlens/checks/_records.py
torchlens/checks/_scan.py
torchlens/checks/_session.py
torchlens/compat/__init__.py
torchlens/compat/_report.py
torchlens/compat/lovely.py
torchlens/compat/torchextractor.py
torchlens/compat/torchshow.py
torchlens/conformance/__init__.py
torchlens/conformance/_adapters.py
torchlens/conformance/_cases.py
torchlens/data_classes/__init__.py
torchlens/data_classes/_accessor_base.py
torchlens/data_classes/_backend_capability_guards.py
torchlens/data_classes/_call_tree.py
torchlens/data_classes/_compaction.py
torchlens/data_classes/_layer_accessors.py
torchlens/data_classes/_layer_sites.py
torchlens/data_classes/_layer_spec.py
torchlens/data_classes/_lookup_keys.py
torchlens/data_classes/_module_role_hints.py
torchlens/data_classes/_nonfinite.py
torchlens/data_classes/_op_dedup.py
torchlens/data_classes/_op_transforms.py
torchlens/data_classes/_preprocessing_provenance.py
torchlens/data_classes/_repr.py
torchlens/data_classes/_runtime_handles.py
torchlens/data_classes/_schema_bindings.py
torchlens/data_classes/_site_key.py
torchlens/data_classes/_state_adapter.py
torchlens/data_classes/_summary.py
torchlens/data_classes/_trace_accessors.py
torchlens/data_classes/_trace_components.py
torchlens/data_classes/_trace_export.py
torchlens/data_classes/_trace_fork.py
torchlens/data_classes/_trace_intervention.py
torchlens/data_classes/_trace_inventory.py
torchlens/data_classes/_trace_legacy_state.py
torchlens/data_classes/_trace_narrate.py
torchlens/data_classes/_trace_profile.py
torchlens/data_classes/_trace_rehydrate.py
torchlens/data_classes/_trace_stack.py
torchlens/data_classes/_trace_stats.py
torchlens/data_classes/_trace_validation.py
torchlens/data_classes/_trace_viz.py
torchlens/data_classes/_value_repr.py
torchlens/data_classes/aten_op.py
torchlens/data_classes/backward_pass.py
torchlens/data_classes/buffer.py
torchlens/data_classes/cleanup.py
torchlens/data_classes/container.py
torchlens/data_classes/derived_grad.py
torchlens/data_classes/field_policy.py
torchlens/data_classes/func_call_location.py
torchlens/data_classes/grad_fn.py
torchlens/data_classes/grad_fn_call.py
torchlens/data_classes/interface.py
torchlens/data_classes/internal_types.py
torchlens/data_classes/layer.py
torchlens/data_classes/module.py
torchlens/data_classes/op.py
torchlens/data_classes/param.py
torchlens/data_classes/prehook.py
torchlens/data_classes/trace.py
torchlens/debug/__init__.py
torchlens/debug/_audit.py
torchlens/debug/_common.py
torchlens/debug/_compile_counter.py
torchlens/debug/_cost.py
torchlens/debug/_determinism.py
torchlens/debug/_dtype_range.py
torchlens/debug/_first_bad.py
torchlens/debug/_flops_vs_dispatch.py
torchlens/debug/_grad_fn_walk.py
torchlens/debug/_gradients.py
torchlens/debug/_graph.py
torchlens/debug/_graph_breaks.py
torchlens/debug/_infer_failure_classify.py
torchlens/debug/_infer_input_shape.py
torchlens/debug/_nan.py
torchlens/debug/_nan_backward.py
torchlens/debug/_params.py
torchlens/debug/_params_audit.py
torchlens/debug/_precision.py
torchlens/debug/_recompute.py
torchlens/debug/_rerun.py
torchlens/distributed/__init__.py
torchlens/distributed/_audit.py
torchlens/distributed/_dtensor.py
torchlens/distributed/_ledger.py
torchlens/distributed/_lifecycle.py
torchlens/distributed/_recognizer.py
torchlens/ecosystem/__init__.py
torchlens/ecosystem/compat.py
torchlens/ecosystem/migrate.py
torchlens/ecosystem/plugins.py
torchlens/errors/__init__.py
torchlens/errors/_base.py
torchlens/errors/episode.py
torchlens/errors/runnable.py
torchlens/examples/__init__.py
torchlens/experiment/__init__.py
torchlens/experiment/_engine.py
torchlens/experiment/_head_sugar.py
torchlens/experiment/_ledger.py
torchlens/experiment/_mcp.py
torchlens/experimental/__init__.py
torchlens/experimental/node_styles.py
torchlens/experimental/dagua/__init__.py
torchlens/experimental/dagua/_bridge.py
torchlens/export/__init__.py
torchlens/export/_common.py
torchlens/export/_graphs.py
torchlens/export/_netron.py
torchlens/export/_netron_attachment.py
torchlens/export/_netron_emit.py
torchlens/export/_netron_fields.py
torchlens/export/_netron_module.py
torchlens/export/_netron_records.py
torchlens/export/_registry.py
torchlens/export/_report.py
torchlens/export/_trackers.py
torchlens/export/_model_explorer/__init__.py
torchlens/export/_model_explorer/_attrs.py
torchlens/export/_model_explorer/_build.py
torchlens/export/_model_explorer/_collection.py
torchlens/export/_model_explorer/_diff.py
torchlens/export/_model_explorer/_edges.py
torchlens/export/_model_explorer/_episode.py
torchlens/export/_model_explorer/_errors.py
torchlens/export/_model_explorer/_files.py
torchlens/export/_model_explorer/_grouprows.py
torchlens/export/_model_explorer/_ids.py
torchlens/export/_model_explorer/_namespace.py
torchlens/export/_model_explorer/_options.py
torchlens/export/_model_explorer/_overlay.py
torchlens/export/_model_explorer/_serve.py
torchlens/export/_model_explorer/_validate.py
torchlens/fastlog/__init__.py
torchlens/fastlog/_halt.py
torchlens/fastlog/_indexes.py
torchlens/fastlog/_record_one_shot.py
torchlens/fastlog/_recorder.py
torchlens/fastlog/_storage_resolver.py
torchlens/fastlog/_validation.py
torchlens/fastlog/cleanup.py
torchlens/fastlog/dry_run.py
torchlens/fastlog/exceptions.py
torchlens/fastlog/options.py
torchlens/fastlog/recover.py
torchlens/fastlog/storage_disk.py
torchlens/fastlog/storage_ram.py
torchlens/fastlog/types.py
torchlens/intervention/__init__.py
torchlens/intervention/_content_identity.py
torchlens/intervention/_helper_fingerprint.py
torchlens/intervention/_metrics.py
torchlens/intervention/_module_alias_guard.py
torchlens/intervention/_module_boundary.py
torchlens/intervention/_replay_context.py
torchlens/intervention/_rerun_predicate.py
torchlens/intervention/_staged_spec_persistence.py
torchlens/intervention/audit.py
torchlens/intervention/binding.py
torchlens/intervention/bundle.py
torchlens/intervention/compose.py
torchlens/intervention/edge_semantics.py
torchlens/intervention/edge_substitution.py
torchlens/intervention/errors.py
torchlens/intervention/handles.py
torchlens/intervention/helpers.py
torchlens/intervention/hooks.py
torchlens/intervention/injection.py
torchlens/intervention/masked_edit.py
torchlens/intervention/model_door.py
torchlens/intervention/param_substitution.py
torchlens/intervention/population.py
torchlens/intervention/predicates.py
torchlens/intervention/regions.py
torchlens/intervention/replay.py
torchlens/intervention/rerun.py
torchlens/intervention/resolver.py
torchlens/intervention/runtime.py
torchlens/intervention/save.py
torchlens/intervention/selectors.py
torchlens/intervention/site_keys.py
torchlens/intervention/sites.py
torchlens/intervention/spec.py
torchlens/intervention/spec_compat.py
torchlens/intervention/steering.py
torchlens/intervention/stochastic.py
torchlens/intervention/sweep.py
torchlens/intervention/types.py
torchlens/intervention/_super/__init__.py
torchlens/intervention/_super/_accessor_base.py
torchlens/intervention/_super/_base.py
torchlens/intervention/_super/super_logs.py
torchlens/intervention/_super/super_op.py
torchlens/intervention/_topology/__init__.py
torchlens/intervention/_topology/topology.py
torchlens/io/__init__.py
torchlens/ir/__init__.py
torchlens/ir/capture_events.py
torchlens/ir/container.py
torchlens/ir/container_registry.py
torchlens/ir/events.py
torchlens/ir/intervention.py
torchlens/ir/live_index.py
torchlens/ir/op_record.py
torchlens/ir/op_record_manifest.py
torchlens/ir/op_record_scatter.py
torchlens/ir/predicate.py
torchlens/ir/predicate_registry.py
torchlens/ir/refs.py
torchlens/ir/selector_eval.py
torchlens/ir/semantics.py
torchlens/ir/summary_role.py
torchlens/ir/workspaces.py
torchlens/mechinterp/__init__.py
torchlens/mechinterp/_aliases.py
torchlens/mechinterp/_anchors.py
torchlens/mechinterp/_compose.py
torchlens/mechinterp/_demo.py
torchlens/mechinterp/_dla.py
torchlens/mechinterp/_errors.py
torchlens/mechinterp/_grids.py
torchlens/mechinterp/_heads.py
torchlens/mechinterp/_lowered.py
torchlens/mechinterp/_params.py
torchlens/mechinterp/_prompts.py
torchlens/mechinterp/_records.py
torchlens/mechinterp/_residual.py
torchlens/mechinterp/_retention.py
torchlens/mechinterp/_scores.py
torchlens/mechinterp/_walk.py
torchlens/mechinterp/translation.json
torchlens/merged/__init__.py
torchlens/merged/_artifact.py
torchlens/merged/_engine.py
torchlens/merged/_enums.py
torchlens/merged/_errors.py
torchlens/merged/_evidence.py
torchlens/merged/_presenter.py
torchlens/neuro/__init__.py
torchlens/neuro/_datasets.py
torchlens/neuro/_handoff.py
torchlens/neuro/_rdms.py
torchlens/notebook/__init__.py
torchlens/notebook/_access.py
torchlens/notebook/_axis_labels.py
torchlens/notebook/_grid.py
torchlens/notebook/_motifs.py
torchlens/notebook/_truncation.py
torchlens/notebook/cards.py
torchlens/notebook/cardtree.py
torchlens/notebook/frontier.py
torchlens/observability/__init__.py
torchlens/observability/_artifact.py
torchlens/observability/_chassis.py
torchlens/observability/_collector.py
torchlens/observability/_errors.py
torchlens/observability/_join.py
torchlens/observability/_kernels.py
torchlens/observability/_kineto.py
torchlens/observability/_memory_parity.py
torchlens/observability/_memory_totals.py
torchlens/observability/_native_profile.py
torchlens/observability/_op_tier.py
torchlens/observability/_overhead.py
torchlens/observability/_payloads.py
torchlens/observability/_perf_harness.py
torchlens/observability/_quantiles.py
torchlens/observability/_region.py
torchlens/observability/_render.py
torchlens/observability/_schema.py
torchlens/observability/_session.py
torchlens/observability/_spans.py
torchlens/observe/__init__.py
torchlens/observe/_device_memory.py
torchlens/observe/_liveness.py
torchlens/observe/_peaks.py
torchlens/observe/_svg.py
torchlens/observe/_timeline.py
torchlens/partial/__init__.py
torchlens/postprocess/__init__.py
torchlens/postprocess/_buffer_addresses.py
torchlens/postprocess/_contracts.py
torchlens/postprocess/_executor.py
torchlens/postprocess/_grouping_stamp.py
torchlens/postprocess/_ingest_contract.py
torchlens/postprocess/_lazy_param_geometry.py
torchlens/postprocess/_materialize.py
torchlens/postprocess/_primitive_profile.py
torchlens/postprocess/_recurrence.py
torchlens/postprocess/_selective_save.py
torchlens/postprocess/_site_join.py
torchlens/postprocess/_site_key.py
torchlens/postprocess/ast_branches.py
torchlens/postprocess/control_flow.py
torchlens/postprocess/finalization.py
torchlens/postprocess/graph_traversal.py
torchlens/postprocess/incremental.py
torchlens/postprocess/labeling.py
torchlens/postprocess/loop_detection.py
torchlens/postprocess/loop_grouping_adapter.py
torchlens/postprocess/saved_summary.py
torchlens/preprocessing/__init__.py
torchlens/preprocessing/_audit.py
torchlens/preprocessing/_authorities.py
torchlens/preprocessing/_diagnostics.py
torchlens/preprocessing/_records.py
torchlens/quickstart/__init__.py
torchlens/quickstart/_gate.py
torchlens/quickstart/_grammar.py
torchlens/quickstart/_ladder.py
torchlens/quickstart/_plan.py
torchlens/quickstart/_primitive.py
torchlens/quickstart/_provenance.py
torchlens/quickstart/_render.py
torchlens/quickstart/_resolve.py
torchlens/receptive_field/__init__.py
torchlens/receptive_field/_engine.py
torchlens/receptive_field/_engine_descriptor.py
torchlens/receptive_field/_engine_forward.py
torchlens/receptive_field/_engine_geometry.py
torchlens/receptive_field/_errors.py
torchlens/receptive_field/_forward_query.py
torchlens/receptive_field/_gradient.py
torchlens/receptive_field/_gradient_forward.py
torchlens/receptive_field/_path.py
torchlens/receptive_field/_query.py
torchlens/receptive_field/_rules.py
torchlens/receptive_field/_table.py
torchlens/receptive_field/_types.py
torchlens/receptive_field/_validation.py
torchlens/receptive_field/_view.py
torchlens/receptive_field/_viz.py
torchlens/receptive_field/rules/__init__.py
torchlens/receptive_field/rules/_utils.py
torchlens/receptive_field/rules/attention.py
torchlens/receptive_field/rules/conv_pool.py
torchlens/receptive_field/rules/elementwise.py
torchlens/receptive_field/rules/interpolation.py
torchlens/receptive_field/rules/linear.py
torchlens/receptive_field/rules/norms.py
torchlens/receptive_field/rules/sequence.py
torchlens/receptive_field/rules/transforms.py
torchlens/repgeom/__init__.py
torchlens/repgeom/_annotation_gate.py
torchlens/repgeom/_geometry.py
torchlens/repgeom/_node_visuals.py
torchlens/repgeom/_trace_views.py
torchlens/report/__init__.py
torchlens/report/_agent_json.py
torchlens/report/_compute_truth.py
torchlens/report/_cost_backward.py
torchlens/report/_cost_perf.py
torchlens/report/_cost_scaling.py
torchlens/report/_cost_tree.py
torchlens/report/_explain.py
torchlens/report/_factcore.py
torchlens/report/_flops_report.py
torchlens/report/_health.py
torchlens/report/_profile.py
torchlens/report/_registry.py
torchlens/report/_report_protocol.py
torchlens/report/_stats_table.py
torchlens/report/_summary_charset.py
torchlens/report/_summary_config.py
torchlens/report/_summary_html.py
torchlens/report/_summary_ladder.py
torchlens/report/_summary_render.py
torchlens/report/_summary_report.py
torchlens/report/_summary_result.py
torchlens/schemas/agent_api_map_v2.json
torchlens/schemas/agent_compare_v1.json
torchlens/schemas/agent_doctor_v1.json
torchlens/schemas/agent_dump_v1.json
torchlens/schemas/agent_envelope_v1.json
torchlens/schemas/agent_error_v1.json
torchlens/schemas/agent_explain_v1.json
torchlens/schemas/agent_overview_folded_v1.json
torchlens/schemas/agent_overview_manifest_v1.json
torchlens/schemas/agent_payload_stats_v1.json
torchlens/schemas/agent_query_sites_v1.json
torchlens/schemas/agent_schema_v1.json
torchlens/schemas/tlspec_manifest_v1.json
torchlens/schemas/tlspec_manifest_v2.json
torchlens/schemas/writer_contract_v9.json
torchlens/semantic/__init__.py
torchlens/semantic/_hops.py
torchlens/semantic/_norm_reconstruction.py
torchlens/semantic/coverage.py
torchlens/semantic/facets.py
torchlens/semantic/logit_lens.py
torchlens/semantic/patching.py
torchlens/semantic/reconstruction.py
torchlens/semantic/tolerances.py
torchlens/semantic/recipes/__init__.py
torchlens/semantic/recipes/_helpers.py
torchlens/semantic/recipes/attention.py
torchlens/semantic/recipes/embedding.py
torchlens/semantic/recipes/lm_head.py
torchlens/semantic/recipes/mlp.py
torchlens/semantic/recipes/norm.py
torchlens/semantic/recipes/residual.py
torchlens/snoop/__init__.py
torchlens/snoop/_entry.py
torchlens/snoop/_errors.py
torchlens/snoop/_event.py
torchlens/snoop/_format.py
torchlens/snoop/_narrate.py
torchlens/snoop/_normalize.py
torchlens/snoop/_session.py
torchlens/snoop/_sink.py
torchlens/snoop/_stats.py
torchlens/stats/__init__.py
torchlens/stats/_aggregate.py
torchlens/stats/_envelope.py
torchlens/stats/_fitted.py
torchlens/stats/_glance.py
torchlens/stats/_seams.py
torchlens/stats/_stats_kernel.py
torchlens/stats/_stats_render.py
torchlens/stats/_streaming.py
torchlens/stats/_tensor_stats.py
torchlens/trackers/__init__.py
torchlens/trackers/_amp.py
torchlens/trackers/_callbacks.py
torchlens/trackers/_errors.py
torchlens/trackers/_graph.py
torchlens/trackers/_protocol.py
torchlens/trackers/_records.py
torchlens/trackers/_sinks.py
torchlens/trackers/_tensorboard.py
torchlens/trackers/_wandb.py
torchlens/trackers/_watch.py
torchlens/transforms/__init__.py
torchlens/transforms/_coerce.py
torchlens/transforms/_context.py
torchlens/transforms/_errors.py
torchlens/transforms/_helpers.py
torchlens/transforms/_kernels.py
torchlens/transforms/_pipeline.py
torchlens/transforms/_pooling.py
torchlens/transforms/_projection.py
torchlens/transforms/_registry.py
torchlens/transforms/_spec.py
torchlens/transforms/_srp.py
torchlens/transforms/_srp_hash.py
torchlens/tviz/__init__.py
torchlens/tviz/_bridge.py
torchlens/tviz/_decomp.py
torchlens/tviz/_errors.py
torchlens/tviz/_extract.py
torchlens/tviz/_metrics.py
torchlens/tviz/_mpl.py
torchlens/tviz/_predictions.py
torchlens/tviz/_receipts.py
torchlens/tviz/_records.py
torchlens/tviz/_render_attention.py
torchlens/tviz/_strip.py
torchlens/tviz/_wording.py
torchlens/utils/__init__.py
torchlens/utils/_callable_safety.py
torchlens/utils/_multipass_access.py
torchlens/utils/_package_frames.py
torchlens/utils/_rng_channels.py
torchlens/utils/_subprocess.py
torchlens/utils/_torch_compat.py
torchlens/utils/_torch_symbols.py
torchlens/utils/_type_new_slot.py
torchlens/utils/_type_sequence_slot.py
torchlens/utils/_uninit_alloc.py
torchlens/utils/alias_footprint.py
torchlens/utils/arg_handling.py
torchlens/utils/collections.py
torchlens/utils/display.py
torchlens/utils/env_flags.py
torchlens/utils/facade.py
torchlens/utils/fail_open.py
torchlens/utils/hashing.py
torchlens/utils/introspection.py
torchlens/utils/lazy_state.py
torchlens/utils/rng.py
torchlens/utils/tensor_utils.py
torchlens/validation/__init__.py
torchlens/validation/_autograd_grad_replay.py
torchlens/validation/_completeness_backstop.py
torchlens/validation/_degenerate_forward.py
torchlens/validation/_destination_coverage.py
torchlens/validation/_edge_boundary.py
torchlens/validation/_index_domain.py
torchlens/validation/_integer_mod_proof.py
torchlens/validation/_invariants_backward_domain.py
torchlens/validation/_invariants_backward_flow.py
torchlens/validation/_invariants_backward_graph.py
torchlens/validation/_invariants_buffers.py
torchlens/validation/_invariants_conditional_base.py
torchlens/validation/_invariants_conditional_modules.py
torchlens/validation/_invariants_conditionals.py
torchlens/validation/_invariants_connectivity.py
torchlens/validation/_invariants_entry.py
torchlens/validation/_invariants_equivalence.py
torchlens/validation/_invariants_modules_params.py
torchlens/validation/_invariants_payloads.py
torchlens/validation/_invariants_primitive_ops.py
torchlens/validation/_invariants_sites.py
torchlens/validation/_invariants_topology.py
torchlens/validation/_layer_grad_report.py
torchlens/validation/_live_model_state.py
torchlens/validation/_non_pass_record.py
torchlens/validation/_norm_zero_weight_proof.py
torchlens/validation/_output_walk.py
torchlens/validation/_pristine.py
torchlens/validation/_replay_device.py
torchlens/validation/_replay_grad_fidelity.py
torchlens/validation/_saved_value_proofs.py
torchlens/validation/_source_provenance.py
torchlens/validation/_stock_layer_grads.py
torchlens/validation/_structural_slot_checks.py
torchlens/validation/_value_predicates.py
torchlens/validation/backward.py
torchlens/validation/consolidated.py
torchlens/validation/core.py
torchlens/validation/diagnostics.py
torchlens/validation/exemptions.py
torchlens/validation/invariants.py
torchlens/validation/status.py
torchlens/visualization/__init__.py
torchlens/visualization/_arg_suppression.py
torchlens/visualization/_backward_inventory.py
torchlens/visualization/_buffer_visibility.py
torchlens/visualization/_bundle_graph.py
torchlens/visualization/_collapse_disclosures.py
torchlens/visualization/_collapse_frontier.py
torchlens/visualization/_collapse_runs.py
torchlens/visualization/_collapse_signatures.py
torchlens/visualization/_condensed_flow.py
torchlens/visualization/_draw_validation.py
torchlens/visualization/_edge_multiplicity.py
torchlens/visualization/_encoding.py
torchlens/visualization/_geometry_audit.py
torchlens/visualization/_label_format.py
torchlens/visualization/_legend.py
torchlens/visualization/_mutated_params.py
torchlens/visualization/_render_common.py
torchlens/visualization/_render_dot.py
torchlens/visualization/_render_edges.py
torchlens/visualization/_render_entrypoints.py
torchlens/visualization/_render_flow.py
torchlens/visualization/_render_leaf.py
torchlens/visualization/_render_nodes.py
torchlens/visualization/_render_ordering.py
torchlens/visualization/_render_regions.py
torchlens/visualization/_render_utils.py
torchlens/visualization/_segment_descriptors.py
torchlens/visualization/_stacking.py
torchlens/visualization/_surgery_diff.py
torchlens/visualization/_svg_compose.py
torchlens/visualization/_typography.py
torchlens/visualization/auto_collapse.py
torchlens/visualization/bundle_diff.py
torchlens/visualization/code_panel.py
torchlens/visualization/collapse_estimator.py
torchlens/visualization/collapse_fallback.py
torchlens/visualization/collapse_ladder.py
torchlens/visualization/collapse_optimizer.py
torchlens/visualization/collapse_patterns.py
torchlens/visualization/collapse_plan.py
torchlens/visualization/fastlog_live.py
torchlens/visualization/fastlog_preview.py
torchlens/visualization/modes.py
torchlens/visualization/node_spec.py
torchlens/visualization/node_universe.py
torchlens/visualization/overlays.py
torchlens/visualization/render_execution.py
torchlens/visualization/render_ir.py
torchlens/visualization/renderer_registry.py
torchlens/visualization/request.py
torchlens/visualization/source_graph.py
torchlens/visualization/surgery_visuals.py
torchlens/visualization/theme_registry.py
torchlens/visualization/themes.py
torchlens/visualization/_rank_layout_internal/__init__.py
torchlens/visualization/_rank_layout_internal/layout.py
torchlens/visualization/_summary_internal/__init__.py
torchlens/visualization/_summary_internal/_builder.py
torchlens/visualization/_summary_internal/_discoverability.py
torchlens/visualization/lenses/__init__.py
torchlens/visualization/lenses/_budget.py
torchlens/visualization/lenses/_families.py
torchlens/visualization/lenses/_filter.py
torchlens/visualization/lenses/_nonfinite.py
torchlens/visualization/lenses/_resolve.py
torchlens/visualization/lenses/_roster.py
torchlens/visualization/lenses/audit/__init__.py
torchlens/visualization/lenses/audit/answer_key.py
torchlens/visualization/lenses/audit/battery.py
torchlens/visualization/lenses/audit/corpus.py
torchlens/visualization/lenses/audit/stage0.py
torchlens/visualization/renderers/__init__.py
torchlens/visualization/renderers/base.py
torchlens/visualization/renderers/graphviz.py
torchlens/viz/__init__.py
torchlens/viz/_heatmap_axes.py
torchlens/viz/_tensor_display.py
torchlens/viz/batch_summary.py
torchlens/viz/feature_maps.py
torchlens/viz/node_plots.py