Computer vision

Important

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

This page provides notebook examples for computer vision tasks using AI Runtime. These examples demonstrate how to train and fine-tune models for various computer vision applications.

Tutorial	Description
Image classification using convolutional neural network	This notebook provides a simple example of how to train a 2-D convolution neural network on serverless GPUs for image classification.
Object detection using RetinaNet	This notebook demonstrates how to train an object detection model using RetinaNet on serverless GPU.
Object detection using YOLO11n	This notebook demonstrates how to train a YOLO11n object detection model on the COCO128 dataset using serverless GPU, with MLflow tracking and Model Serving deployment.