Skip to main content

Distributed training using Ray

Beta

This feature is in Beta.

This page has notebook examples for distributed training using Ray on Serverless GPU compute.

Train PyTorch ResNet model using Ray

This notebook demonstrates distributed training of a PyTorch ResNet model on the FashionMNIST dataset using Ray Train and Ray Data based on the Serverless GPU Distributed Python API.

Notebook

Open notebook in new tab