LICENSE
README.md
pyproject.toml
src/examples/tracing_example.py
src/examples/tracing_on_bert.py
src/examples/tracing_on_cnn.py
src/examples/tracing_on_llama.py
src/examples/benchmark/bert_ddp.py
src/examples/benchmark/vit_ddp.py
src/traceml/__init__.py
src/traceml/cli.py
src/traceml/config.py
src/traceml/decorators.py
src/traceml/distributed.py
src/traceml/runner.py
src/traceml/runtime.py
src/traceml/session.py
src/traceml/stdout_stderr_capture.py
src/traceml/tcp_transport.py
src/traceml/database/__init__.py
src/traceml/database/database.py
src/traceml/database/database_sender.py
src/traceml/database/database_writer.py
src/traceml/database/remote_database_store.py
src/traceml/loggers/__init__.py
src/traceml/loggers/error_log.py
src/traceml/renderers/__init__.py
src/traceml/renderers/base_renderer.py
src/traceml/renderers/layer_combined_memory_renderer.py
src/traceml/renderers/layer_combined_timing_renderer.py
src/traceml/renderers/model_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/utils.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/page_layout.py
src/traceml/renderers/display/nicegui_sections/helper.py
src/traceml/renderers/display/nicegui_sections/layer_memory_table_section.py
src/traceml/renderers/display/nicegui_sections/layer_timer_table_section.py
src/traceml/renderers/display/nicegui_sections/main_page.py
src/traceml/renderers/display/nicegui_sections/model_combined_section.py
src/traceml/renderers/display/nicegui_sections/process_section.py
src/traceml/renderers/display/nicegui_sections/steptiming_section.py
src/traceml/renderers/display/nicegui_sections/system_section.py
src/traceml/renderers/layer_combined_memory/__init__.py
src/traceml/renderers/layer_combined_memory/services.py
src/traceml/renderers/layer_combined_timing/__init__.py
src/traceml/renderers/layer_combined_timing/services.py
src/traceml/samplers/__init__.py
src/traceml/samplers/base_sampler.py
src/traceml/samplers/layer_backward_memory_sampler.py
src/traceml/samplers/layer_backward_time_sampler.py
src/traceml/samplers/layer_forward_memory_sampler.py
src/traceml/samplers/layer_forward_time_sampler.py
src/traceml/samplers/layer_memory_sampler.py
src/traceml/samplers/model_forward_memory_sampler.py
src/traceml/samplers/process_sampler.py
src/traceml/samplers/stdout_stderr_sampler.py
src/traceml/samplers/step_memory_sampler.py
src/traceml/samplers/steptimer_sampler.py
src/traceml/samplers/system_sampler.py
src/traceml/samplers/schema/process.py
src/traceml/samplers/schema/system.py
src/traceml/utils/__init__.py
src/traceml/utils/base_trace_event.py
src/traceml/utils/cuda_event_pool.py
src/traceml/utils/dataloader_patch.py
src/traceml/utils/entry_hook.py
src/traceml/utils/flush_buffers.py
src/traceml/utils/formatting.py
src/traceml/utils/layer_backward_memory_hook.py
src/traceml/utils/layer_backward_time_hooks.py
src/traceml/utils/layer_forward_memory_hook.py
src/traceml/utils/layer_forward_time_hooks.py
src/traceml/utils/layer_parameter_memory.py
src/traceml/utils/model_forward_memory_hook.py
src/traceml/utils/shared_utils.py
src/traceml/utils/step_memory.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