AI/ML API

This section documents the AI/ML specific tracing features.

DFTracerAI Class

AI Module

The ai module provides decorators and utilities for tracing AI/ML workflows.

AI Categories

ProfileCategory

AI Components

Data Operations

Dataloader Operations

Device Operations

Compute Operations

Communication Operations

Checkpoint Operations

Pipeline Operations

AI Instance

Utility Functions

Constants

Module Instances

Convenience instances for common operations:

ai - Main AI tracing instance
data - Data operations
dataloader - Dataloader operations
device - Device transfer operations
compute - Compute operations
comm - Communication operations
checkpoint - Checkpoint operations
pipeline - Pipeline operations