AutoGluon-Cloud

Train and Deploy AutoGluon in the Cloud

AutoGluon is an open-source AutoML library that trains state-of-the-art ML models on tabular, time-series, and multimodal data with just a few lines of code. AutoGluon-Cloud takes that same API and runs it on AWS — train models and serve predictions on Amazon SageMaker without managing infrastructure or setting up a heavyweight ML environment on your local machine.

It supports two workflows:

• Train your own predictor — the same fit → deploy → predict workflow as local AutoGluon, with all the heavy lifting offloaded to SageMaker.

• Run pretrained foundation models — deploy state-of-the-art pretrained models like Chronos-2 for zero-shot inference, with no training required.

Train AutoGluon predictors in the cloud

Full walkthrough: Tabular, Time Series.

Tabular

Train a classification or regression model on tabular data.

from autogluon.cloud import TabularCloudPredictor

# `train_data` and `test_data` can be a local path, S3 URL, or pandas DataFrame
train_data = "https://autogluon.s3.amazonaws.com/datasets/Inc/train.csv"
test_data = "https://autogluon.s3.amazonaws.com/datasets/Inc/test.csv"

# Train
cloud_predictor = TabularCloudPredictor()
cloud_predictor.fit(
    train_data=train_data,
    predictor_init_args={"label": "class"},  # passed to TabularPredictor()
    predictor_fit_args={"time_limit": 120},  # passed to TabularPredictor.fit()
)

# Real-time inference endpoint
cloud_predictor.deploy()
result = cloud_predictor.predict_real_time(test_data)
cloud_predictor.cleanup_deployment()

# Batch prediction
result = cloud_predictor.predict(test_data)

Time Series

Forecast future values of time series.

from autogluon.cloud import TimeSeriesCloudPredictor

# `data` can be a local path, S3 URL, or pandas DataFrame
data = "https://autogluon.s3.amazonaws.com/datasets/timeseries/m4_hourly_tiny/train.csv"

# Train
cloud_predictor = TimeSeriesCloudPredictor()
cloud_predictor.fit(
    train_data=data,
    predictor_init_args={"target": "target", "prediction_length": 24},  # passed to TimeSeriesPredictor()
    predictor_fit_args={"time_limit": 120},  # passed to TimeSeriesPredictor.fit()
)

# Real-time inference endpoint
cloud_predictor.deploy()
result = cloud_predictor.predict_real_time(data)
cloud_predictor.cleanup_deployment()

# Batch prediction
result = cloud_predictor.predict(data)

Run pretrained foundation models

Full walkthrough: Time Series.

Time Series (Chronos-2)

Zero-shot forecasts with a pretrained model — no training required.

from autogluon.cloud import TimeSeriesFoundationModel

# `data` can be a local path, S3 URL, or pandas DataFrame
data = "https://autogluon.s3.amazonaws.com/datasets/timeseries/m4_hourly_tiny/train.csv"

model = TimeSeriesFoundationModel("chronos-2")

# Batch prediction
predictions = model.predict(
    data=data,
    target="target",
    prediction_length=24,
)

# Real-time inference endpoint
endpoint = model.deploy()
predictions = endpoint.predict(
    data=data,
    target="target",
    prediction_length=24,
)
endpoint.delete_endpoint()

Installation

pip install autogluon.cloud

Before running the examples above, set up your AWS resources (IAM role + S3 bucket) by following the Setup tutorial.