AutoGluon-Cloud

Train and Deploy AutoGluon in the Cloud

AutoGluon-Cloud makes it easy to run AutoGluon in the cloud. With a few lines of code, you can train models and run inference on Amazon SageMaker — without managing infrastructure or installing AutoGluon’s heavy dependencies on your local machine.

It supports two workflows:

• Train AutoGluon predictors in the cloud — 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: Train Your Own Models

Tabular

Train a classification or regression model on tabular data.

from autogluon.cloud import TabularCloudPredictor

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 = "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: Use Foundation Models

Time Series (Chronos-2)

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

from autogluon.cloud import TimeSeriesFoundationModel

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.