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.
- Apache Spark MLlib
- Export and import ML models
- Third-party machine learning integrations