Deep learning based recommender systems

Important

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

These notebooks build deep learning recommender systems on AI Runtime. They show how to create a two-tower recommendation model using PyTorch Lightning for distributed training.

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