D

AI Runtime

Infrastructure service

by Databricks

AI Runtime provides serverless GPU compute for training, fine-tuning and batch inference of custom models.

Find alternatives to AI Runtime

What it does

• GPU cloud
• Model training
• Model inference

Features 8

• Notebooks on serverless GPUAttach a notebook to serverless GPU compute and develop interactively with no cluster setup.
• IDE over SSHConnect from an IDE or terminal over an SSH tunnel to work on a GPU node directly.
• Databricks CLISubmit and manage distributed GPU training jobs from a laptop using YAML job configs.
• AI agent skillUses the databricks-ai-runtime agent skill to submit, monitor and manage GPU training jobs from an AI agent.
• Ray supportRuns Ray Core, Ray Data, Ray Train and Ray Tune on serverless GPU compute.
• Pre-built environmentsStart from a minimal Standard environment or a Databricks AI environment pre-loaded with ML frameworks.
• Efficient data loadersStreams data from Unity Catalog volumes with fault-tolerant loaders built for high GPU utilization.
• Logs viewerShows training output and GPU resource usage while code runs.

Sources

Pricing

Pricing page · 15 Oct 2026

"Only pay for the products you use at per second granularity."

Description

Official documentation · 15 Oct 2026

"train and fine-tune custom models"

What it does

Official documentation · 15 Oct 2026

"Run large-scale batch inference"

Sold within

Official documentation · 15 Oct 2026

"brings GPU support for Databricks Serverless"

Name

Official documentation · 15 Oct 2026

"AI Runtime is a serverless GPU compute offering"

URL

Official documentation · 15 Oct 2026

"AI Runtime is a serverless GPU compute offering"

Deployment

Official documentation · 15 Oct 2026

"Fully managed GPU infrastructure: Serverless"

Interfaces

Official documentation · 18 Oct 2026

Page: 'Databricks CLI Submit and manage distributed GPU training jobs from your laptop using YAML job configs'.

Also from Databricks

• Agent Bricks Agent platform

  A control plane for building, evaluating, governing and monitoring AI agents across proprietary and open-source models, with native MCP support.

• Databricks AI Search Data service

  A managed hybrid semantic, keyword and vector search service (formerly Mosaic AI Vector Search) with automatic data sync and Unity Catalog governance.

• Databricks Model Serving Model API

  A managed endpoint service for deploying and governing classical ML models, generative models and agents from one interface.

• Databricks Model Training Developer tool

  A managed service for fine-tuning open-source LLMs or training custom models on enterprise data using dedicated GPU infrastructure.

• Databricks AI Platform

  Databricks' tooling for building, fine-tuning, serving and evaluating models and agents on lakehouse data.

• Databricks Data + AI Platform Platform

  Lakehouse platform for storing, governing, querying and sharing enterprise data.

• AI/BI Application

  AI-native business-intelligence product comprising AI/BI Dashboards (AI-assisted dashboard/visualization creation) and Genie Spaces (conversational natural-language exploration of data), built into the Databricks Data + AI Platform with Unity Catalog governance and no per-seat licensing.

• Genie One Application

  AI coworker (the evolution of the earlier Databricks Assistant/Genie) that lets business users ask questions, take action and drive outcomes over enterprise data via natural language, integrating with Slack, Teams, Jira, Google Drive and Salesforce, with mobile apps.

• Unity Gateway Infrastructure service

  Enterprise AI gateway providing centralized cost tracking/budgets, model access (Claude, GPT, Gemini, Grok and others), security/governance (access policies, PII/PHI filtering, audit trails), smart routing and observability across an organization's AI systems.

• Document Intelligence Data service

  Document Intelligence uses AI functions to parse, extract, and classify enterprise documents.

• Databricks Feature Store Data service

  Databricks Feature Store provides a governed layer for building, sharing, and serving machine learning features.

• Genie Agents AI agent

  Genie Agents answer natural-language data questions with SQL queries, result tables, and visualizations.

• AI/BI Dashboards Application

  AI/BI Dashboards provides AI-assisted authoring, data visualization and forecasting.

Open Databricks in the directory

Something wrong here? Send a correction — quote this product id: databricks-ai-runtime.