Guides
GPUMemoryProfiler
MemorySnapshot
ProfileResult
profile_context()
profile_function()
MemoryVisualizer
MemoryAnalyzer
GapFinding
MemoryTracker
OOMFlightRecorder
OOMFlightRecorderConfig
OOMExceptionClassification
classify_oom_exception()
TelemetryEventV2
DeviceMemoryCollector
DeviceMemorySample
build_device_memory_collector()
detect_torch_runtime_backend()
CPUMemoryProfiler
CPUMemoryTracker
telemetry_event_from_record()
telemetry_event_to_dict()
validate_telemetry_record()
load_telemetry_events()
resolve_distributed_identity()
TimelineMarker
derive_timeline_markers()
derive_session_timeline_markers()
timeline_marker_to_dict()
get_gpu_info()
format_bytes()
convert_bytes()
JAXMemoryProfiler
MemoryWatchdog
JAXProfiler
ProfiledFunction
get_device_info()
get_system_info()
Generated protocol buffer code.