Train and Deploy AutoGluon in the Cloud

AutoGluon-Cloud Documentation | AutoGluon Documentation

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.

💾 Installation & setup

pip install autogluon.cloud

Then provision the IAM role and S3 bucket AutoGluon-Cloud needs on AWS:

from autogluon.cloud import bootstrap

bootstrap()

See the Setup tutorial for the full walkthrough, including how to register an existing role and bucket instead.

⚙️ Train your own model

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)

🚀 Run a pretrained foundation model

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 — no training required
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()