Skip to main content

Deep learning based recommender systems

Public Preview

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 building recommendation systems using AI Runtime. These examples demonstrate how to create efficient recommendation models using modern deep learning approaches.

Tutorial

Description

Two-tower recommendation model

Learn how to create a two-tower recommendation model using PyTorch Lightning for distributed training.