AI Runtime CLI command reference

Important

The AI Runtime CLI is in Beta.

The air CLI provides five subcommands for managing training workloads: air run, air get run, air list runs, air logs, and air cancel. The same reference is always available from the CLI itself:

air --help              # All commands
air <command> --help    # Flags for a specific command
air -h config           # YAML config reference
air -h config.compute   # Per-field help for a YAML section

The CLI help reflects the exact version you have installed, so it is the source of truth if it differs from the table below.

Commands

Command	Purpose
air run	Submit a workload defined by a YAML file. Supports --file, --watch, --dry-run, --override, --idempotency-key, --email, and -p to select an authentication profile.
air get run	Show metadata, status, and configuration for a specific run.
air list runs	List recent runs. Use --limit to bound the result count and --active to show only running workloads.
air logs	Stream or download logs for a run. Use --node to target a specific node and --download-to to write logs to a directory.
air cancel	Cancel a running workload.
air register image	Register and cache a custom Docker image for use in workloads. See Use custom Docker images.

Global flags

Flag	Purpose
--version	Print the installed CLI version.
-p, --profile	Use the named Databricks CLI authentication profile instead of the default.
-h, --help	Show help. When followed by a config path (-h config.compute), show YAML field help.

See also

• AI Runtime CLI quickstart
• Workload YAML reference
• Track runs with MLflow and the Jobs run page
• Use custom Docker images