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)

© Copyright 2026, Databricks.

Built with Sphinx using a theme provided by Read the Docs.