Skip to main content

Classic machine learning

Beta

This feature is in Beta.

This page provides notebook examples for classic machine learning tasks using Serverless GPU compute. These examples demonstrate how to leverage GPUs for traditional ML algorithms and time series forecasting.

Before running these notebooks, see the Best practices checklist.

XGBoost model training

This notebook demonstrates how to train an XGBoost regression model on a single GPU. XGBoost can significantly benefit from GPU acceleration for large datasets.

Notebook

Open notebook in new tab

Time series forecasting with GluonTS

This notebook demonstrates an end-to-end workflow for probabilistic time-series forecasting of electricity consumption data with GluonTS's DeepAR model on a serverless GPU cluster. It covers data ingestion, resampling, model training, prediction, visualization, and evaluation.

Notebook

Open notebook in new tab