LICENSE
README.md
pyproject.toml
tests/test_api_renames.py
tests/test_arg_positions.py
tests/test_ast_branches.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_func_config.py
tests/test_gc.py
tests/test_internals.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_log.py
tests/test_output_aesthetics.py
tests/test_param_log.py
tests/test_profiling.py
tests/test_real_world_models.py
tests/test_save_new_activations.py
tests/test_summary.py
tests/test_toy_models.py
tests/test_validation.py
torchlens/__init__.py
torchlens/_deprecations.py
torchlens/_literals.py
torchlens/_state.py
torchlens/constants.py
torchlens/options.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/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/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/postprocess/__init__.py
torchlens/postprocess/ast_branches.py
torchlens/postprocess/control_flow.py
torchlens/postprocess/finalization.py
torchlens/postprocess/graph_traversal.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/core.py
torchlens/validation/exemptions.py
torchlens/validation/invariants.py
torchlens/visualization/__init__.py
torchlens/visualization/dagua_bridge.py
torchlens/visualization/elk_layout.py
torchlens/visualization/rendering.py