Serverless GPU Python API

Serverless GPU API is a lightweight, easy-to-use library that simplifies running multi-GPU workloads on Databricks AI Runtime, directly from Databricks notebooks.

It provides the same functionality as running torchrun train.py or accelerate launch train.py from the command line, but integrates seamlessly into the Databricks notebook environment.

For concepts, usage guides, and examples, see the AI Runtime documentation.

• API Reference
• Environment Variables

Indices and tables

• Index

• Module Index

• Search Page