LICENSE
README.md
pyproject.toml
src/examples/tracing_on_bert.py
src/examples/tracing_small_example.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/config.py
src/traceml/decorators.py
src/traceml/session.py
src/traceml/tracer.py
src/traceml/database/__init__.py
src/traceml/database/database.py
src/traceml/database/database_writer.py
src/traceml/loggers/__init__.py
src/traceml/loggers/error_log.py
src/traceml/manager/__init__.py
src/traceml/manager/tracker_manager.py
src/traceml/renderers/__init__.py
src/traceml/renderers/activation_gradient_memory_renderer.py
src/traceml/renderers/base_renderer.py
src/traceml/renderers/layer_combined_renderer.py
src/traceml/renderers/process_renderer.py
src/traceml/renderers/stdout_stderr_renderer.py
src/traceml/renderers/steptimer_renderer.py
src/traceml/renderers/system_renderer.py
src/traceml/renderers/display/__init__.py
src/traceml/renderers/display/cli_display_manager.py
src/traceml/renderers/display/layout.py
src/traceml/renderers/display/nicegui_display_manager.py
src/traceml/renderers/display/notebook_display_manager.py
src/traceml/renderers/display/stdout_stderr_capture.py
src/traceml/renderers/display/nicegui_sections/helper.py
src/traceml/renderers/display/nicegui_sections/layer_table_section.py
src/traceml/renderers/display/nicegui_sections/main_page.py
src/traceml/renderers/display/nicegui_sections/process_section.py
src/traceml/renderers/display/nicegui_sections/system_section.py
src/traceml/renderers/layer_combined/__init__.py
src/traceml/renderers/layer_combined/services.py
src/traceml/renderers/layer_combined/utils.py
src/traceml/samplers/__init__.py
src/traceml/samplers/activation_memory_sampler.py
src/traceml/samplers/activation_time_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/steptimer_sampler.py
src/traceml/samplers/system_sampler.py
src/traceml/utils/__init__.py
src/traceml/utils/activation_memory_hook.py
src/traceml/utils/activation_time_hooks.py
src/traceml/utils/formatting.py
src/traceml/utils/gradient_hook.py
src/traceml/utils/patch.py
src/traceml/utils/steptimer.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