LICENSE
README.md
pyproject.toml
tests/test_metadata.py
tests/test_real_world_models.py
tests/test_real_world_models_slow.py
tests/test_toy_models.py
torchlens/__init__.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/model_history.py
torchlens/postprocess.py
torchlens/tensor_log.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