Skip to main content

Train and Deploy AutoGluon in the Cloud

PyPI Python Versions GitHub license Continuous Integration

AutoGluon-Cloud Documentation | AutoGluon Documentation

AutoGluon-Cloud lets you train and deploy state-of-the-art ML models in the cloud in a few lines of code. Run AutoGluon on Amazon SageMaker without worrying about infrastructure, dependencies, or a heavy local ML environment. It supports two workflows:

💾 Installation & setup

pip install autogluon.cloud

Then provision the IAM role and S3 bucket AutoGluon-Cloud needs to run 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

Train an AutoGluon predictor on your data and serve it from a SageMaker endpoint — same API as local AutoGluon, all heavy lifting on AWS. Full walkthrough: tabular, time series.

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

Skip training entirely — deploy a pretrained model like Chronos-2 to SageMaker and get zero-shot predictions out of the box. Full walkthrough: time series.

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()

Metadata

Release files for autogluon.cloud 0.6.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for autogluon.cloud 0.6.0
File Size Uploaded
autogluon_cloud-0.6.0.tar.gz 109.1 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for autogluon.cloud 0.6.0
File Interpreter ABI Platform
autogluon_cloud-0.6.0-py3-none-any.whl Python 3 none any Details

Total release size: 247.9 kB

Release files / autogluon_cloud-0.6.0.tar.gz

Download URL autogluon_cloud-0.6.0.tar.gz
Size 109.1 kB
Tags Source
SHA-256 checksum
How to use checksums
b3842372cb7ef796cb9cda10d4b2d44cceec6496787e9732b2fd922ff4354f4c
BLAKE2b-256 checksum
How to use checksums
21fab003256f55733971919033ea1c64dda96244c2721e8a9b2f1a502ce4de4b
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.11.16

Release files / autogluon_cloud-0.6.0-py3-none-any.whl

Download URL autogluon_cloud-0.6.0-py3-none-any.whl
Size 138.8 kB
Tags Python 3
SHA-256 checksum
How to use checksums
c64339cf3577e7b87c600e6506c6a25c0f4566ac6f39c596536d5456e4fa21a1
BLAKE2b-256 checksum
How to use checksums
6ca849b3059a595738390ccb891eab049162a4ecd38de7c1e92b3a1b32f63b44
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.11.16

Release history Release notifications | RSS feed

This release

0.6.0 This release

2 release files

0.5.0

2 release files

0.4.2

2 release files

0.4.1

2 release files

0.4.0

2 release files

0.3.1

2 release files

0.3.0

2 release files

0.2.0

2 release files

0.1

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page