Machine Learning

Accelerate your AI projects with a data-centric approach to machine learning

Try for free Schedule a demo


Built on an open lakehouse architecture, Databricks Machine Learning empowers ML teams to prepare and process data, streamlines cross-team collaboration and standardizes the full ML lifecycle from experimentation to production.

MLflow prediction metrics Diagram

Simplify all aspects of data for ML

Because Databricks ML is built on an open lakehouse foundation with Delta Lake, you can empower your machine learning teams to access, explore and prepare any type of data, from batch or streaming pipelines, at any scale. Turn features into production pipelines in a self-service manner without depending on data engineering support.

Automate experiment tracking and governance

Managed MLflow automatically tracks your experiments and logs parameters, metrics, versioning of data and code, as well as model artifacts with each training run. You can quickly see previous runs, compare results and reproduce a past result, as needed. Once you have identified the best version of a model for production, register it to the Model Registry to simplify handoffs along the deployment lifecycle.

Automate  experiment example

Activities example

Manage the full model lifecycle from data to production — and back

Once trained models are registered, you can collaboratively manage them through their lifecycle with the Model Registry. Models can be versioned and moved through various stages, like experimentation, staging, production and archived. The lifecycle management integrates with approval and governance workflows according to role-based access controls. Comments and email notifications provide a rich collaborative environment for data teams.

Deploy ML models at scale and low latency

Deploy models with a single click without having to worry about server management or scale constraints. With Databricks, you can deploy your models as REST API endpoints anywhere with enterprise-grade availability.

MLflow Model registry Diagram


Product components

Migrate to Databricks

Tired of the data silos, slow performance and high costs associated with legacy systems like Hadoop and enterprise data warehouses? Migrate to the Databricks Lakehouse: the modern platform for all your data, analytics and AI use cases.

Migrate to Databricks


Customer Story

AI is transforming drug discovery

AstraZeneca leverages data and NLP to help scientists research novel drugs


Customer Story

Reinventing mobile banking with ML

How HSBC manages cashless transactions for over 39M customers with ML-based fraud detection


Customer Story

Brewing data and AI at scale

Starbucks serves up omnichannel experiences across 30,000 stores with Databricks


Customer Story

The future of entertainment with AI

Comcast delivers an Emmy-winning viewer experience with AI-powered voice remotes


All the resources you need. All in one place.  Image

All the resources you need. All in one place.

Explore the resource library to find eBooks and videos on data science and machine learning topics.

Ready to get

Try Databricks for free