LICENSE
README.md
pyproject.toml
tests/test_activation_postfunc_refactor.py
tests/test_api_renames.py
tests/test_arg_positions.py
tests/test_ast_branches.py
tests/test_autograd_saved_bytes.py
tests/test_backward.py
tests/test_backward_streaming.py
tests/test_backward_visualization.py
tests/test_buffer_visibility.py
tests/test_code_panel.py
tests/test_collapse_fn.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_decoration.py
tests/test_defaults.py
tests/test_extra_data_plumbing.py
tests/test_fastlog_postfunc_parity.py
tests/test_func_config.py
tests/test_gc.py
tests/test_internals.py
tests/test_introspection.py
tests/test_io_bundle.py
tests/test_io_corruption.py
tests/test_io_export.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_large_graphs.py
tests/test_layer_log.py
tests/test_metadata.py
tests/test_module_focus.py
tests/test_module_log.py
tests/test_multi_trace.py
tests/test_node_modes.py
tests/test_node_spec_api.py
tests/test_output_aesthetics.py
tests/test_param_log.py
tests/test_perf_bundle.py
tests/test_profiling.py
tests/test_real_world_models.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_skip_fn.py
tests/test_summary.py
tests/test_toy_models.py
tests/test_two_pass_inplace_fix.py
tests/test_validation.py
torchlens/__init__.py
torchlens/_deprecations.py
torchlens/_errors.py
torchlens/_literals.py
torchlens/_robustness.py
torchlens/_state.py
torchlens/_training_validation.py
torchlens/constants.py
torchlens/options.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/_summary/__init__.py
torchlens/_summary/_builder.py
torchlens/capture/__init__.py
torchlens/capture/arg_positions.py
torchlens/capture/backward.py
torchlens/capture/flops.py
torchlens/capture/output_tensors.py
torchlens/capture/salient_args.py
torchlens/capture/source_tensors.py
torchlens/capture/tensor_tracking.py
torchlens/capture/trace.py
torchlens/data_classes/__init__.py
torchlens/data_classes/_lookup_keys.py
torchlens/data_classes/_summary.py
torchlens/data_classes/buffer_log.py
torchlens/data_classes/cleanup.py
torchlens/data_classes/func_call_location.py
torchlens/data_classes/grad_fn_log.py
torchlens/data_classes/grad_fn_pass_log.py
torchlens/data_classes/interface.py
torchlens/data_classes/internal_types.py
torchlens/data_classes/layer_log.py
torchlens/data_classes/layer_pass_log.py
torchlens/data_classes/model_log.py
torchlens/data_classes/module_log.py
torchlens/data_classes/param_log.py
torchlens/decoration/__init__.py
torchlens/decoration/model_prep.py
torchlens/decoration/torch_funcs.py
torchlens/fastlog/__init__.py
torchlens/fastlog/_indexes.py
torchlens/fastlog/_orchestrator.py
torchlens/fastlog/_predicate.py
torchlens/fastlog/_record_context.py
torchlens/fastlog/_record_one_shot.py
torchlens/fastlog/_recorder.py
torchlens/fastlog/_state.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/multi_trace/__init__.py
torchlens/multi_trace/bundle.py
torchlens/multi_trace/metrics.py
torchlens/multi_trace/node_view.py
torchlens/multi_trace/topology.py
torchlens/postprocess/__init__.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/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/backward.py
torchlens/validation/core.py
torchlens/validation/exemptions.py
torchlens/validation/invariants.py
torchlens/visualization/__init__.py
torchlens/visualization/code_panel.py
torchlens/visualization/dagua_bridge.py
torchlens/visualization/elk_layout.py
torchlens/visualization/fastlog_live.py
torchlens/visualization/fastlog_preview.py
torchlens/visualization/modes.py
torchlens/visualization/node_spec.py
torchlens/visualization/rendering.py