A2ML is distributed as a python package. To install the client with default AugerAI AutoML provider.

$ pip install -U a2ml

Provider Groups

Azure AutoML Provider

To include the Azure AutoML provider.

$ pip install "a2ml[azure]"


Azure requires that LightGBM be installed.

$ brew install lightgbm

To run Azure AutoML models locally.


This will install scikit-learn and several additional dependencies.

$ pip install "a2ml[azure]"

Google Cloud Provider

To include the Google Cloud AutoML provider.

$ pip install "a2ml[google]"

Server Installation

To install server dependencies.

$ pip install "a2ml[server]"

Install Everything

To install all providers and server dependencies.

$ pip install "a2ml[all]"