LICENSE
NOTICE
README.md
pyproject.toml
tests/test_activation_postfunc_refactor.py
tests/test_animation.py
tests/test_api_renames.py
tests/test_api_surface.py
tests/test_api_surface_deprecation.py
tests/test_append_semantics.py
tests/test_append_streaming.py
tests/test_append_validator.py
tests/test_arg_kwargs_extraction.py
tests/test_arg_positions.py
tests/test_ast_branches.py
tests/test_auto_capture.py
tests/test_autograd_saved_bytes.py
tests/test_autoroute_registry.py
tests/test_backend_registry.py
tests/test_backward.py
tests/test_backward_attribution_unchanged.py
tests/test_backward_intervention.py
tests/test_backward_p1_triggers.py
tests/test_backward_p2_projections.py
tests/test_backward_p3_payloads.py
tests/test_backward_p5_modules.py
tests/test_backward_p7_validation.py
tests/test_backward_streaming.py
tests/test_backward_visualization.py
tests/test_bridges_launch.py
tests/test_bridges_stretch.py
tests/test_buffer_datamodel.py
tests/test_buffer_sole_output.py
tests/test_buffer_visibility.py
tests/test_bug_fixes_phase14.py
tests/test_bundle_alignment.py
tests/test_bundle_diff_renderer.py
tests/test_bundle_methods.py
tests/test_capture_events_parity.py
tests/test_capture_targets.py
tests/test_capture_unification_p0.py
tests/test_capture_unification_p1.py
tests/test_capture_unification_p2.py
tests/test_capture_unification_p3a.py
tests/test_capture_unification_p3b.py
tests/test_capture_unification_p4.py
tests/test_capture_unification_p5.py
tests/test_capture_unification_p6.py
tests/test_capture_unification_p7.py
tests/test_capture_unification_p8.py
tests/test_check_plan_link_consistency.py
tests/test_code_panel.py
tests/test_collapse_fn.py
tests/test_combined_visualization.py
tests/test_compat_report.py
tests/test_conditional_branches.py
tests/test_conditional_integration.py
tests/test_conditional_invariants.py
tests/test_conditional_lifecycle.py
tests/test_conditional_multipass.py
tests/test_conditional_rendering.py
tests/test_conditional_step5.py
tests/test_cross_file_refactor.py
tests/test_dagua_theme.py
tests/test_debug_utils.py
tests/test_decoration.py
tests/test_defaults.py
tests/test_device_context_logging.py
tests/test_docs_snippets.py
tests/test_doctor.py
tests/test_draw_headless.py
tests/test_edge_counts.py
tests/test_examples.py
tests/test_examples_load.py
tests/test_exception_taxonomy.py
tests/test_export_html_minimal.py
tests/test_exports.py
tests/test_extra_data_plumbing.py
tests/test_extractor_compat.py
tests/test_fastlog_backward.py
tests/test_fastlog_exception_in_predicate.py
tests/test_fastlog_halt.py
tests/test_fastlog_memory_parity.py
tests/test_fastlog_postfunc_parity.py
tests/test_fastlog_train_mode_grad.py
tests/test_field_lifecycle_matrix.py
tests/test_format_helpers.py
tests/test_freeze_module.py
tests/test_func_config.py
tests/test_gc.py
tests/test_glossary_conformance_b2b.py
tests/test_glossary_conformance_b2c.py
tests/test_glossary_v5_additions.py
tests/test_gradient_stats.py
tests/test_input_coercion.py
tests/test_internals.py
tests/test_intervention_error_catalog.py
tests/test_intervention_phase10.py
tests/test_intervention_phase11.py
tests/test_intervention_phase12.py
tests/test_intervention_phase13.py
tests/test_intervention_phase2.py
tests/test_intervention_phase3.py
tests/test_intervention_phase4_5.py
tests/test_intervention_phase4a.py
tests/test_intervention_phase4b.py
tests/test_intervention_phase4c.py
tests/test_intervention_phase5.py
tests/test_intervention_phase6.py
tests/test_intervention_phase7.py
tests/test_intervention_phase8a.py
tests/test_intervention_phase8b.py
tests/test_intervention_phase9.py
tests/test_intervention_tensor_replacement.py
tests/test_introspection.py
tests/test_io_bundle.py
tests/test_io_corruption.py
tests/test_io_export.py
tests/test_io_format_v3_load.py
tests/test_io_integration.py
tests/test_io_lazy.py
tests/test_io_pandas.py
tests/test_io_pickle.py
tests/test_io_plain_pickle.py
tests/test_io_scrub_policy.py
tests/test_io_security.py
tests/test_io_streaming.py
tests/test_ir_basic.py
tests/test_jax_backend_capture.py
tests/test_jax_backend_derived_grads.py
tests/test_jax_backend_hardening.py
tests/test_label_geometry.py
tests/test_large_graphs.py
tests/test_layer_grad_oracle.py
tests/test_layer_log.py
tests/test_loop_module_rolling.py
tests/test_metadata.py
tests/test_migrations.py
tests/test_misc_capture.py
tests/test_mlx_backend_smoke.py
tests/test_mlx_hardening.py
tests/test_module_containment_equality.py
tests/test_module_containment_save_load.py
tests/test_module_focus.py
tests/test_module_log.py
tests/test_module_stack_buffer_ordering.py
tests/test_module_stack_exception_safety.py
tests/test_module_stack_identity_exhaustive.py
tests/test_module_stack_user_hook_ordering.py
tests/test_multi_output_modules.py
tests/test_multi_pass.py
tests/test_multi_trace_followons.py
tests/test_nan_debugging.py
tests/test_no_legacy_tl_attrs.py
tests/test_node_modes.py
tests/test_node_spec_api.py
tests/test_not_mvp_audit.py
tests/test_observers_backward.py
tests/test_onramp.py
tests/test_options.py
tests/test_options_wiring.py
tests/test_output_aesthetics.py
tests/test_overlays.py
tests/test_packaging_diet.py
tests/test_param_log.py
tests/test_perf_bundle.py
tests/test_perf_gate.py
tests/test_profiling.py
tests/test_quantities_memory.py
tests/test_real_world_models.py
tests/test_record_export.py
tests/test_report_explain.py
tests/test_reprs.py
tests/test_robustness_pr1.py
tests/test_robustness_pr2.py
tests/test_robustness_pr3.py
tests/test_robustness_pr4.py
tests/test_save_new_activations.py
tests/test_schema_audit.py
tests/test_selector_unification_phase4.py
tests/test_sequential_hooks.py
tests/test_session.py
tests/test_sites.py
tests/test_skip_fn.py
tests/test_stop_early.py
tests/test_streaming_stats.py
tests/test_subfolder_imports.py
tests/test_summary.py
tests/test_tap_record_span.py
tests/test_themes.py
tests/test_tinygrad_backend_capture.py
tests/test_tinygrad_backend_validation.py
tests/test_tl_lifecycle.py
tests/test_tl_meta.py
tests/test_tlspec_backcompat.py
tests/test_tlspec_detection.py
tests/test_tlspec_intervention_transitional.py
tests/test_tlspec_unified.py
tests/test_torch_func_transforms.py
tests/test_toy_models.py
tests/test_trace_autoroute.py
tests/test_trace_autoroute_image.py
tests/test_trace_autoroute_input_preprocessor.py
tests/test_trace_autoroute_multimodal.py
tests/test_trace_field_invariant.py
tests/test_trace_halt.py
tests/test_two_pass_inplace_fix.py
tests/test_utility_validation_runner_regressions.py
tests/test_validate_consolidated.py
tests/test_validation.py
torchlens/__init__.py
torchlens/_debug.py
torchlens/_deprecations.py
torchlens/_errors.py
torchlens/_input_coerce.py
torchlens/_literals.py
torchlens/_robustness.py
torchlens/_source_links.py
torchlens/_state.py
torchlens/_trace_state.py
torchlens/_training_validation.py
torchlens/captured_run.py
torchlens/constants.py
torchlens/observers.py
torchlens/options.py
torchlens/quantities.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/requires.txt
torchlens.egg-info/top_level.txt
torchlens/_io/__init__.py
torchlens/_io/accessor_rebuild.py
torchlens/_io/bundle.py
torchlens/_io/lazy.py
torchlens/_io/manifest.py
torchlens/_io/paths.py
torchlens/_io/rehydrate.py
torchlens/_io/scrub.py
torchlens/_io/streaming.py
torchlens/_io/tensor_policy.py
torchlens/_io/tlspec.py
torchlens/_summary/__init__.py
torchlens/accessors/__init__.py
torchlens/autoroute/__init__.py
torchlens/autoroute/_builtin_input.py
torchlens/autoroute/_registry.py
torchlens/autoroute/input/__init__.py
torchlens/autoroute/output/__init__.py
torchlens/backends/__init__.py
torchlens/backends/_protocol.py
torchlens/backends/default_specs.py
torchlens/backends/registry.py
torchlens/backends/jax/__init__.py
torchlens/backends/jax/backend.py
torchlens/backends/jax/capabilities.py
torchlens/backends/jax/jaxpr.py
torchlens/backends/mlx/__init__.py
torchlens/backends/mlx/backend.py
torchlens/backends/mlx/capabilities.py
torchlens/backends/mlx/model_prep.py
torchlens/backends/mlx/tensor_store.py
torchlens/backends/mlx/wrappers.py
torchlens/backends/tinygrad/__init__.py
torchlens/backends/tinygrad/backend.py
torchlens/backends/tinygrad/capabilities.py
torchlens/backends/torch/__init__.py
torchlens/backends/torch/_tl.py
torchlens/backends/torch/aliasing.py
torchlens/backends/torch/backend.py
torchlens/backends/torch/backward.py
torchlens/backends/torch/buffer_writes.py
torchlens/backends/torch/model_prep.py
torchlens/backends/torch/module_stack.py
torchlens/backends/torch/ops.py
torchlens/backends/torch/sources.py
torchlens/backends/torch/tensor_tracking.py
torchlens/backends/torch/wrappers.py
torchlens/bridge/__init__.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/lit.py
torchlens/bridge/nnsight.py
torchlens/bridge/profiler.py
torchlens/bridge/repeng.py
torchlens/bridge/rsatoolbox.py
torchlens/bridge/sae_lens.py
torchlens/bridge/shap.py
torchlens/bridge/steering_vectors.py
torchlens/callbacks/__init__.py
torchlens/callbacks/lightning.py
torchlens/capture/__init__.py
torchlens/capture/arg_positions.py
torchlens/capture/flops.py
torchlens/capture/predicates.py
torchlens/capture/projections.py
torchlens/capture/salient_args.py
torchlens/capture/trace.py
torchlens/compat/__init__.py
torchlens/compat/_report.py
torchlens/compat/lovely.py
torchlens/compat/torchextractor.py
torchlens/compat/torchshow.py
torchlens/data_classes/__init__.py
torchlens/data_classes/_accessor_base.py
torchlens/data_classes/_lookup_keys.py
torchlens/data_classes/_module_role_hints.py
torchlens/data_classes/_summary.py
torchlens/data_classes/backward_pass.py
torchlens/data_classes/buffer.py
torchlens/data_classes/cleanup.py
torchlens/data_classes/derived_grad.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/trace.py
torchlens/debug/__init__.py
torchlens/errors/__init__.py
torchlens/errors/_base.py
torchlens/examples/__init__.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/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/grad/__init__.py
torchlens/intervene/__init__.py
torchlens/intervention/__init__.py
torchlens/intervention/_metrics.py
torchlens/intervention/bundle.py
torchlens/intervention/errors.py
torchlens/intervention/handles.py
torchlens/intervention/helpers.py
torchlens/intervention/hooks.py
torchlens/intervention/predicates.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/sites.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/buffer.py
torchlens/ir/events.py
torchlens/ir/intervention.py
torchlens/ir/live_index.py
torchlens/ir/predicate.py
torchlens/ir/refs.py
torchlens/ir/semantics.py
torchlens/llm/__init__.py
torchlens/neuro/__init__.py
torchlens/notebook/__init__.py
torchlens/paper/__init__.py
torchlens/partial/__init__.py
torchlens/postprocess/__init__.py
torchlens/postprocess/_materialize.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/report/__init__.py
torchlens/report/_explain.py
torchlens/semantic/__init__.py
torchlens/semantic/facets.py
torchlens/semantic/patching.py
torchlens/semantic/reconstruction.py
torchlens/semantic/recipes/__init__.py
torchlens/semantic/recipes/_helpers.py
torchlens/semantic/recipes/attention.py
torchlens/semantic/recipes/embedding.py
torchlens/semantic/recipes/mlp.py
torchlens/semantic/recipes/norm.py
torchlens/semantic/recipes/residual.py
torchlens/stats/__init__.py
torchlens/utils/__init__.py
torchlens/utils/arg_handling.py
torchlens/utils/collections.py
torchlens/utils/display.py
torchlens/utils/hashing.py
torchlens/utils/introspection.py
torchlens/utils/rng.py
torchlens/utils/tensor_utils.py
torchlens/validation/__init__.py
torchlens/validation/_layer_grad_report.py
torchlens/validation/_stock_layer_grads.py
torchlens/validation/backward.py
torchlens/validation/consolidated.py
torchlens/validation/core.py
torchlens/validation/exemptions.py
torchlens/validation/invariants.py
torchlens/viewer/__init__.py
torchlens/visualization/__init__.py
torchlens/visualization/_label_format.py
torchlens/visualization/_render_utils.py
torchlens/visualization/bundle_diff.py
torchlens/visualization/code_panel.py
torchlens/visualization/fastlog_live.py
torchlens/visualization/fastlog_preview.py
torchlens/visualization/modes.py
torchlens/visualization/node_spec.py
torchlens/visualization/overlays.py
torchlens/visualization/rendering.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/viz/__init__.py
torchlens/viz/_tensor_display.py
torchlens/viz/batch_summary.py