pub mod document;
pub mod events;
pub mod health;
pub mod memory;
pub mod schema;
pub use document::{parse_trace, write_jsonl, TraceDocument};
pub use events::{
DeviceMemoryEvent, EdgeEvent, GradientEvent, MemoryEvent, OpEvent, SpanEndEvent,
SpanStartEvent, TensorEvent, TraceEvent,
};
pub use health::{analyze_health, EvidenceCoverage, HealthIssue, HealthSeverity, TraceHealth};
pub use memory::{category_for_step, storage_bytes, MemoryAction, MemoryCategory};
pub use schema::{
GradientState, SpanKind, SpanRecord, TimingMode, TraceRunMeta, TraceSummary, SCHEMA,
};