NVIDIA
NVIDIA
DLRM for PyTorch
Resource
NVIDIA
NVIDIA
DLRM for PyTorch

The Deep Learning Recommendation Model (DLRM) is a recommendation model designed to make use of both categorical and numerical inputs.

Changelog

April 2020

  • Initial release

May 2020

  • Performance optimizations

June 2020

  • Updated performance tables to include A100 results and multi-GPU setup
  • Multi-GPU optimizations

August 2020

  • Preprocessing with Spark 3 on GPU
  • Multiple performance optimizations
  • Automatic placement and load balancing of embedding tables
  • Improved README

Known issues

There are no known issues with this model