LICENSE
README.md
pyproject.toml
src/examples/basic_example.py
src/examples/ddp_example.py
src/examples/fsdp_gpu_example.py
src/examples/hf-trainer-minimal.py
src/examples/input-stall.py
src/examples/straggler_ddp_example.py
src/examples/advanced/bert_ddp.py
src/examples/advanced/bert_gradient_accum.py
src/examples/advanced/bert_memory_creep.py
src/examples/advanced/bert_straggler.py
src/examples/advanced/cnn_mnist.py
src/examples/advanced/hf_trainer_integration.py
src/examples/advanced/hf_trainer_vision.py
src/examples/advanced/llama_finetuning.py
src/examples/advanced/test_lightning_traceml.py
src/examples/advanced/vit_ddp.py
src/traceml/__init__.py
src/traceml/cli.py
src/traceml/decorators.py
src/traceml/aggregator/__init__.py
src/traceml/aggregator/aggregator_main.py
src/traceml/aggregator/final_summary.py
src/traceml/aggregator/sqlite_writer.py
src/traceml/aggregator/trace_aggregator.py
src/traceml/aggregator/display_drivers/__init__.py
src/traceml/aggregator/display_drivers/base.py
src/traceml/aggregator/display_drivers/cli.py
src/traceml/aggregator/display_drivers/layout.py
src/traceml/aggregator/display_drivers/nicegui.py
src/traceml/aggregator/display_drivers/page_layout.py
src/traceml/aggregator/display_drivers/nicegui_sections/__init__.py
src/traceml/aggregator/display_drivers/nicegui_sections/helper.py
src/traceml/aggregator/display_drivers/nicegui_sections/layer_memory_table_section.py
src/traceml/aggregator/display_drivers/nicegui_sections/layer_timer_table_section.py
src/traceml/aggregator/display_drivers/nicegui_sections/model_combined_section.py
src/traceml/aggregator/display_drivers/nicegui_sections/model_diagnostics_section.py
src/traceml/aggregator/display_drivers/nicegui_sections/pages.py
src/traceml/aggregator/display_drivers/nicegui_sections/process_section.py
src/traceml/aggregator/display_drivers/nicegui_sections/step_memory_section.py
src/traceml/aggregator/display_drivers/nicegui_sections/steptiming_section.py
src/traceml/aggregator/display_drivers/nicegui_sections/system_section.py
src/traceml/aggregator/display_drivers/nicegui_sections/ui_shell.py
src/traceml/aggregator/sqlite_writers/__init__.py
src/traceml/aggregator/sqlite_writers/process.py
src/traceml/aggregator/sqlite_writers/step_memory.py
src/traceml/aggregator/sqlite_writers/step_time.py
src/traceml/aggregator/sqlite_writers/system.py
src/traceml/aggregator/summaries/__init__.py
src/traceml/aggregator/summaries/process.py
src/traceml/aggregator/summaries/step_time.py
src/traceml/aggregator/summaries/step_time_diagnosis.py
src/traceml/aggregator/summaries/system.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/diagnostics/__init__.py
src/traceml/diagnostics/common.py
src/traceml/diagnostics/model_diagnostics.py
src/traceml/diagnostics/step_memory.py
src/traceml/diagnostics/step_memory_formatters.py
src/traceml/diagnostics/step_memory_trend.py
src/traceml/diagnostics/step_time.py
src/traceml/diagnostics/step_time_formatters.py
src/traceml/diagnostics/step_time_trend.py
src/traceml/diagnostics/trends.py
src/traceml/integrations/__init__.py
src/traceml/integrations/huggingface.py
src/traceml/integrations/lightning.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/stdout_stderr_renderer.py
src/traceml/renderers/user_time_renderer.py
src/traceml/renderers/utils.py
src/traceml/renderers/layer_combined_memory/__init__.py
src/traceml/renderers/layer_combined_memory/compute.py
src/traceml/renderers/layer_combined_memory/renderer.py
src/traceml/renderers/layer_combined_memory/schema.py
src/traceml/renderers/layer_combined_time/__init__.py
src/traceml/renderers/layer_combined_time/compute.py
src/traceml/renderers/layer_combined_time/renderer.py
src/traceml/renderers/layer_combined_time/schema.py
src/traceml/renderers/model_diagnostics/renderer.py
src/traceml/renderers/process/__init__.py
src/traceml/renderers/process/cli_compute.py
src/traceml/renderers/process/common.py
src/traceml/renderers/process/computer.py
src/traceml/renderers/process/dashboard_compute.py
src/traceml/renderers/process/renderer.py
src/traceml/renderers/step_memory/cli_compute.py
src/traceml/renderers/step_memory/common.py
src/traceml/renderers/step_memory/computer.py
src/traceml/renderers/step_memory/dashboard_compute.py
src/traceml/renderers/step_memory/diagnostics.py
src/traceml/renderers/step_memory/renderer.py
src/traceml/renderers/step_memory/schema.py
src/traceml/renderers/step_time/__init__.py
src/traceml/renderers/step_time/compute.py
src/traceml/renderers/step_time/diagnostics.py
src/traceml/renderers/step_time/renderer.py
src/traceml/renderers/step_time/schema.py
src/traceml/renderers/system/__init__.py
src/traceml/renderers/system/cli_compute.py
src/traceml/renderers/system/common.py
src/traceml/renderers/system/computer.py
src/traceml/renderers/system/dashboard_compute.py
src/traceml/renderers/system/renderer.py
src/traceml/runtime/__init__.py
src/traceml/runtime/config.py
src/traceml/runtime/executor.py
src/traceml/runtime/runtime.py
src/traceml/runtime/session.py
src/traceml/runtime/settings.py
src/traceml/runtime/stdout_stderr_capture.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/step_time_sampler.py
src/traceml/samplers/system_sampler.py
src/traceml/samplers/schema/layer_forward_backward_memory.py
src/traceml/samplers/schema/layer_forward_backward_time.py
src/traceml/samplers/schema/layer_memory.py
src/traceml/samplers/schema/process.py
src/traceml/samplers/schema/step_memory.py
src/traceml/samplers/schema/step_time_schema.py
src/traceml/samplers/schema/system.py
src/traceml/transport/__init__.py
src/traceml/transport/distributed.py
src/traceml/transport/tcp_transport.py
src/traceml/utils/__init__.py
src/traceml/utils/base_trace_event.py
src/traceml/utils/cuda_event_pool.py
src/traceml/utils/entry_hook.py
src/traceml/utils/flush_buffers.py
src/traceml/utils/formatting.py
src/traceml/utils/layer_parameter_memory.py
src/traceml/utils/shared_utils.py
src/traceml/utils/step_memory.py
src/traceml/utils/timing.py
src/traceml/utils/ast_analysis/__init__.py
src/traceml/utils/ast_analysis/code_manifest.py
src/traceml/utils/ast_analysis/helpers.py
src/traceml/utils/ast_analysis/models.py
src/traceml/utils/ast_analysis/scanner.py
src/traceml/utils/ast_analysis/visitor.py
src/traceml/utils/hooks/__init__.py
src/traceml/utils/hooks/layer_backward_memory_hook.py
src/traceml/utils/hooks/layer_backward_time_hooks.py
src/traceml/utils/hooks/layer_forward_memory_hook.py
src/traceml/utils/hooks/layer_forward_time_hooks.py
src/traceml/utils/hooks/model_forward_memory_hook.py
src/traceml/utils/hooks/optimizer_hook.py
src/traceml/utils/patches/__init__.py
src/traceml/utils/patches/backward_auto_timer_patch.py
src/traceml/utils/patches/dataloader_patch.py
src/traceml/utils/patches/forward_auto_timer_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
tests/test_grad_accum.py
tests/test_hf_trainer.py
tests/test_msgpack_roundtrip.py
tests/test_seq_counter.py