AI Runtime example notebooks

Public Preview

AI Runtime for single-node tasks is in Public Preview. The distributed training API for multi-GPU workloads remain in Beta.

The pages below include various notebook examples that show how to use AI Runtime for different tasks.

• • Large language models (LLMs)
  • Examples for fine-tuning large language models including parameter-efficient methods.
• • Computer vision
  • Examples for computer vision tasks including object detection and image classification.
• • Deep learning based recommender systems
  • Examples for building recommendation systems using modern deep learning approaches like two-tower models.
• • Classic ML
  • Examples for traditional machine learning tasks including XGBoost model training and time series forecasting.
• • Multi-GPU distributed training
  • Examples for scaling training across multiple GPUs and nodes using the Serverless GPU API.