Databricks AI Runtime SDK
API Reference
Data Loading and Checkpointing
Environment Variables
Databricks AI Runtime SDK
Index
Index
A
|
C
|
D
|
E
|
G
|
L
|
R
|
S
|
U
|
V
A
AcceleratorType (class in databricks.air.compute)
C
Checkpointable (class in databricks.air.data)
D
DataLoader (class in databricks.air.data)
distributed() (in module databricks.air)
E
environment variable
DATABRICKS_SERVERLESS_GPU_LOG_VERBOSE
G
GPU_1xA10 (databricks.air.compute.AcceleratorType attribute)
GPU_1xH100 (databricks.air.compute.AcceleratorType attribute)
GPU_8xB300 (databricks.air.compute.AcceleratorType attribute)
GPU_8xH100 (databricks.air.compute.AcceleratorType attribute)
L
load_state_dict() (databricks.air.data.Checkpointable method)
(databricks.air.data.DataLoader method)
(databricks.air.data.UCVolumeDataset method)
R
ray_init() (in module databricks.air)
S
state_dict() (databricks.air.data.Checkpointable method)
(databricks.air.data.DataLoader method)
(databricks.air.data.UCVolumeDataset method)
U
UCVolumeDataset (class in databricks.air.data)
UCVolumeReader (class in databricks.air.data)
UCVolumeWriter (class in databricks.air.data)
V
validate_checkpoint_id() (databricks.air.data.UCVolumeReader class method)
(databricks.air.data.UCVolumeWriter class method)