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 for classification, regression, and time series forecasting on Amazon SageMaker. All it takes is a few lines of code; AutoGluon-Cloud handles the infrastructure, dependencies, and glue code for you.

💡 Why AutoGluon-Cloud?

  • Works like local AutoGluon. Pass in DataFrames, get predictions back — as convenient as working locally, with the compute handled by AWS.
  • No boilerplate. No training scripts, inference handlers, or serialization code to write and maintain.
  • Official AWS containers. Everything runs in the AutoGluon Deep Learning Containers, maintained and security-patched by AWS.
  • Sensible defaults, fully configurable. Under the hood it's just SageMaker running in your AWS account, so you stay in full control.

💾 Installation

pip install autogluon.cloud
autogluon-cloud bootstrap  # one-time setup for IAM role and S3 bucket

See the Setup tutorial for more details.

🚀 Foundation models

Zero-shot forecasts with a pretrained model like Chronos-2 — no training required. Full walkthrough.

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

⚙️ Train your own predictor

Train an AutoGluon predictor on your data and serve it from SageMaker. Full walkthrough: time series, tabular.

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

# Batch prediction
result = cloud_predictor.predict(test_data)

# Real-time inference endpoint
endpoint = cloud_predictor.deploy()
result = endpoint.predict(test_data)
endpoint.delete_endpoint()

Metadata

Release files for autogluon.cloud 0.7.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.7.0
File Size Uploaded
autogluon_cloud-0.7.0.tar.gz 102.5 kB Details

Built distribution (wheel)

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

Total release size: 230.0 kB

Release files / autogluon_cloud-0.7.0.tar.gz

Download URL autogluon_cloud-0.7.0.tar.gz
Size 102.5 kB
Tags Source
SHA-256 checksum
How to use checksums
2c5c5b4605bb19fdcad18c33e63fb84fc54a20385444d6c9738c61e533c9c66b
BLAKE2b-256 checksum
How to use checksums
1e4daaea17084d59c62e06673df4d25f3b206a65d71dd38a2bfd0a3c9ae673a7
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.11.17

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

Download URL autogluon_cloud-0.7.0-py3-none-any.whl
Size 127.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
b924b0f31fdd5d30017387d369dbd6dc56a4459b9f78dea90f9409e3d4b6c1eb
BLAKE2b-256 checksum
How to use checksums
5507ed2686551085de98b1e8c40da3d36b40dddfa2b9fa7263b714ed20a6440b
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.11.17

Release history Release notifications | RSS feed

This release

0.7.0 This release

2 release files

0.6.0

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