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:
- Train your own predictor — the same
fit → deploy → predictworkflow 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 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()
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file autogluon_cloud-0.5.0.tar.gz.
File metadata
- Download URL: autogluon_cloud-0.5.0.tar.gz
- Upload date:
- Size: 101.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.11.15
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
84845de964301802c58c5f10710ae29cf82b0863b96c8ead1be35de24deaa4b0
|
|
| MD5 |
42123ed6d11b246665a4e1d9c06124d8
|
|
| BLAKE2b-256 |
e35be3f71424abab536793ca54ed374c5d2fc3e6c80b162ac59d932dd99d0b1f
|
File details
Details for the file autogluon_cloud-0.5.0-py3-none-any.whl.
File metadata
- Download URL: autogluon_cloud-0.5.0-py3-none-any.whl
- Upload date:
- Size: 128.3 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.11.15
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
4858e588ef28cddd44899b2ee6db9791acd940bafd71c1f1bea8376c8c989e94
|
|
| MD5 |
939d94727c809db26a6d7505d1413e7a
|
|
| BLAKE2b-256 |
c3ce12c6720b02c68764c1a9a6c098133de995ca3b254336cbf3e323c850e418
|
