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
- Export and Import ML Models
- Third-Party Machine Learning Integrations