LICENSE
README.md
pyproject.toml
src/examples/tracing_on_bert.py
src/examples/tracing_with_class_decorator.py
src/examples/tracing_with_model_instance.py
src/tests/test_activation_memory_sampler.py
src/tests/test_gradient_memory_sampler.py
src/tests/test_layer_memory_sampler.py
src/tests/test_process_sampler.py
src/tests/test_pytorch_model_sampler.py
src/tests/test_system_sampler.py
src/traceml/__init__.py
src/traceml/cli.py
src/traceml/decorator.py
src/traceml/loggers/__init__.py
src/traceml/loggers/error_log.py
src/traceml/loggers/stdout/__init__.py
src/traceml/loggers/stdout/activation_gradient_memory_logger.py
src/traceml/loggers/stdout/base_stdout_logger.py
src/traceml/loggers/stdout/layer_combined_stdout_logger.py
src/traceml/loggers/stdout/system_process_logger.py
src/traceml/loggers/stdout/display/__init__.py
src/traceml/loggers/stdout/display/cli_display_manager.py
src/traceml/loggers/stdout/display/notebook_display_manager.py
src/traceml/manager/__init__.py
src/traceml/manager/tracker_manager.py
src/traceml/samplers/__init__.py
src/traceml/samplers/activation_memory_sampler.py
src/traceml/samplers/base_sampler.py
src/traceml/samplers/gradient_memory_sampler.py
src/traceml/samplers/layer_memory_sampler.py
src/traceml/samplers/process_sampler.py
src/traceml/samplers/system_sampler.py
src/traceml/utils/__init__.py
src/traceml/utils/activation_hook.py
src/traceml/utils/formatting.py
src/traceml/utils/gradient_hook.py
src/traceml/utils/patch.py
src/traceml_ai.egg-info/PKG-INFO
src/traceml_ai.egg-info/SOURCES.txt
src/traceml_ai.egg-info/dependency_links.txt
src/traceml_ai.egg-info/entry_points.txt
src/traceml_ai.egg-info/requires.txt
src/traceml_ai.egg-info/top_level.txt