LICENSE
README.md
pyproject.toml
tests/test_decoration.py
tests/test_internals.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_toy_models.py
tests/test_validation.py
torchlens/__init__.py
torchlens/_state.py
torchlens/constants.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/capture/__init__.py
torchlens/capture/flops.py
torchlens/capture/output_tensors.py
torchlens/capture/source_tensors.py
torchlens/capture/tensor_tracking.py
torchlens/capture/trace.py
torchlens/data_classes/__init__.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/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/rendering.py