NGC Catalog
CLASSIC
Welcome Guest
Resources
DLRM PyTorch codebase

DLRM PyTorch codebase

For downloads and more information, please view on a desktop device.
Logo for DLRM PyTorch codebase
Description
PyTorch codebase for training and using DLRM model
Publisher
NVIDIA Deep Learning Examples
Latest Version
21.10.1
Modified
April 6, 2023
Compressed Size
1.88 MB

DLRM PyTorch codebase

PyTorch codebase for training and using DLRM model. Refer to the following guides for the use-cases of this codebase, setup instructions and performance numbers:

  • DLRM for PyTorch
  • DLRM Triton deployment for PyTorch