LICENSE
README.md
pyproject.toml
src/dev/scenarios/bert_ddp.py
src/dev/scenarios/bert_memory_creep.py
src/dev/scenarios/bert_straggler.py
src/dev/scenarios/cnn_mnist_deep.py
src/dev/scenarios/hf_trainer_deep_nlp.py
src/dev/scenarios/vit_ddp_stress.py
src/traceml/__init__.py
src/traceml/api.py
src/traceml/cli.py
src/traceml/decorators.py
src/traceml/aggregator/__init__.py
src/traceml/aggregator/aggregator_main.py
src/traceml/aggregator/sqlite_writer.py
src/traceml/aggregator/summary_service.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/summary.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/stdout_stderr.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/analytics/__init__.py
src/traceml/analytics/trends/__init__.py
src/traceml/analytics/trends/core.py
src/traceml/analytics/trends/schema.py
src/traceml/core/__init__.py
src/traceml/core/lifecycle.py
src/traceml/core/registry.py
src/traceml/core/rendering.py
src/traceml/core/summaries.py
src/traceml/core/telemetry.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/registry.py
src/traceml/diagnostics/step_memory_formatters.py
src/traceml/diagnostics/step_memory_trend.py
src/traceml/diagnostics/trends.py
src/traceml/diagnostics/process/__init__.py
src/traceml/diagnostics/process/api.py
src/traceml/diagnostics/process/context.py
src/traceml/diagnostics/process/rules.py
src/traceml/diagnostics/step_memory/__init__.py
src/traceml/diagnostics/step_memory/adapters.py
src/traceml/diagnostics/step_memory/api.py
src/traceml/diagnostics/step_memory/context.py
src/traceml/diagnostics/step_memory/rules.py
src/traceml/diagnostics/step_memory/trend.py
src/traceml/diagnostics/step_memory_summary/__init__.py
src/traceml/diagnostics/step_memory_summary/api.py
src/traceml/diagnostics/step_memory_summary/context.py
src/traceml/diagnostics/step_memory_summary/rules.py
src/traceml/diagnostics/step_time/__init__.py
src/traceml/diagnostics/step_time/api.py
src/traceml/diagnostics/step_time/context.py
src/traceml/diagnostics/step_time/formatters.py
src/traceml/diagnostics/step_time/rules.py
src/traceml/diagnostics/step_time/trend.py
src/traceml/diagnostics/system/__init__.py
src/traceml/diagnostics/system/api.py
src/traceml/diagnostics/system/context.py
src/traceml/diagnostics/system/rules.py
src/traceml/instrumentation/__init__.py
src/traceml/instrumentation/hooks/__init__.py
src/traceml/instrumentation/hooks/layer_backward_memory_hooks.py
src/traceml/instrumentation/hooks/layer_backward_time_hooks.py
src/traceml/instrumentation/hooks/layer_forward_memory_hooks.py
src/traceml/instrumentation/hooks/layer_forward_time_hooks.py
src/traceml/instrumentation/hooks/model_forward_memory_hooks.py
src/traceml/instrumentation/hooks/optimizer_hooks.py
src/traceml/instrumentation/patches/__init__.py
src/traceml/instrumentation/patches/backward_auto_timer_patch.py
src/traceml/instrumentation/patches/dataloader_patch.py
src/traceml/instrumentation/patches/forward_auto_timer_patch.py
src/traceml/integrations/__init__.py
src/traceml/integrations/huggingface.py
src/traceml/integrations/lightning.py
src/traceml/launcher/__init__.py
src/traceml/launcher/cli.py
src/traceml/launcher/commands.py
src/traceml/launcher/manifest.py
src/traceml/launcher/process.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/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/stdout_stderr/common.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/formatter.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/reporting/__init__.py
src/traceml/reporting/final.py
src/traceml/reporting/compare/__init__.py
src/traceml/reporting/compare/command.py
src/traceml/reporting/compare/core.py
src/traceml/reporting/compare/formatters.py
src/traceml/reporting/compare/io.py
src/traceml/reporting/compare/policy.py
src/traceml/reporting/compare/render.py
src/traceml/reporting/compare/verdict.py
src/traceml/reporting/sections/__init__.py
src/traceml/reporting/sections/base.py
src/traceml/reporting/sections/process/__init__.py
src/traceml/reporting/sections/process/builder.py
src/traceml/reporting/sections/process/formatter.py
src/traceml/reporting/sections/process/loader.py
src/traceml/reporting/sections/step_memory/__init__.py
src/traceml/reporting/sections/step_memory/builder.py
src/traceml/reporting/sections/step_memory/formatter.py
src/traceml/reporting/sections/step_memory/loader.py
src/traceml/reporting/sections/step_time/__init__.py
src/traceml/reporting/sections/step_time/builder.py
src/traceml/reporting/sections/step_time/formatter.py
src/traceml/reporting/sections/step_time/loader.py
src/traceml/reporting/sections/system/__init__.py
src/traceml/reporting/sections/system/builder.py
src/traceml/reporting/sections/system/formatter.py
src/traceml/reporting/sections/system/loader.py
src/traceml/reporting/summaries/__init__.py
src/traceml/reporting/summaries/diagnosis_presentation.py
src/traceml/reporting/summaries/issue_summary.py
src/traceml/reporting/summaries/process.py
src/traceml/reporting/summaries/step_memory.py
src/traceml/reporting/summaries/step_time.py
src/traceml/reporting/summaries/step_time_diagnosis.py
src/traceml/reporting/summaries/summary_formatting.py
src/traceml/reporting/summaries/summary_io.py
src/traceml/reporting/summaries/summary_layout.py
src/traceml/reporting/summaries/system.py
src/traceml/runtime/__init__.py
src/traceml/runtime/config.py
src/traceml/runtime/executor.py
src/traceml/runtime/launch_context.py
src/traceml/runtime/runtime.py
src/traceml/runtime/sampler_registry.py
src/traceml/runtime/sender.py
src/traceml/runtime/session.py
src/traceml/runtime/settings.py
src/traceml/runtime/state.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_common.py
src/traceml/samplers/layer_memory_sampler.py
src/traceml/samplers/layer_time_common.py
src/traceml/samplers/process_sampler.py
src/traceml/samplers/runtime_context.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_manifest.py
src/traceml/samplers/system_sampler.py
src/traceml/samplers/utils.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/sdk/__init__.py
src/traceml/sdk/decorators_compat.py
src/traceml/sdk/initial.py
src/traceml/sdk/instrumentation.py
src/traceml/sdk/protocol.py
src/traceml/sdk/summary_client.py
src/traceml/sdk/wrappers.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/msgpack_codec.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_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_base_renderer_contracts.py
tests/test_compare_missing.py
tests/test_core_protocols.py
tests/test_core_registry.py
tests/test_diagnostic_registry.py
tests/test_diagnostics_common.py
tests/test_diagnostics_rule_units.py
tests/test_final_report_generator.py
tests/test_grad_accum.py
tests/test_hf_trainer.py
tests/test_init_and_wrappers.py
tests/test_initialization_and_wrappers.py
tests/test_instrumentation_namespace.py
tests/test_launcher_modules.py
tests/test_msgpack_roundtrip.py
tests/test_runtime_executor.py
tests/test_sampler_registry.py
tests/test_seq_counter.py
tests/test_step_memory_diagnostics_package.py
tests/test_step_memory_formatter.py
tests/test_step_memory_summary_section.py
tests/test_step_time_summary.py
tests/test_summary_section_fixtures.py
tests/test_system_process_sections.py
tests/test_telemetry_publisher.py
tests/test_trace_session_state.py
tests/test_trend_core.py