LICENSE
README.md
pyproject.toml
tests/test_decoration.py
tests/test_metadata.py
tests/test_module_log.py
tests/test_output_aesthetics.py
tests/test_param_log.py
tests/test_real_world_models.py
tests/test_real_world_models_slow.py
tests/test_toy_models.py
torchlens/__init__.py
torchlens/_state.py
torchlens/cleanup.py
torchlens/constants.py
torchlens/decorate_torch.py
torchlens/flops.py
torchlens/helper_funcs.py
torchlens/interface.py
torchlens/logging_funcs.py
torchlens/model_funcs.py
torchlens/trace_model.py
torchlens/user_funcs.py
torchlens/validation.py
torchlens/vis.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/data_classes/__init__.py
torchlens/data_classes/func_call_location.py
torchlens/data_classes/model_log.py
torchlens/data_classes/module_log.py
torchlens/data_classes/param_log.py
torchlens/data_classes/tensor_log.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