This section describes machine learning capabilities in Databricks.
For machine learning workloads, Databricks provides Databricks Runtime for Machine Learning (Databricks Runtime ML), a ready-to-go environment for machine learning and data science. It contains multiple popular libraries, including TensorFlow, PyTorch, Keras, and XGBoost. It also supports distributed training using Horovod. For information on Databricks Runtime ML, see Databricks Runtime for Machine Learning.
- Apache Spark MLlib
- Exporting and Importing ML Models
- Third-Party Machine Learning Integrations