Deploying a Multi-Stage Recommender System#

We created two Jupyter notebooks that demonstrate two different stages of recommender systems. The notebooks show how to deploy a multi-stage recommender system and serve recommendations with Triton Inference Server. The notebooks demonstrate how to use the NVTabular, Merlin Models, and Merlin Systems libraries for feature engineering, training, and then inference.

The two example notebooks are structured as follows:

  • Building the Recommender System:

    • Execute the preprocessing and feature engineering pipeline (ETL) with NVTabular on the GPU/CPU.

    • Train a ranking and retrieval model with TensorFlow based on the ETL output.

    • Export the saved models, user and item features, and item embeddings.

  • Deploying the Recommender System with Triton:

    • Set up a Feast feature store for feature storing and a Faiss index for similarity search.

    • Build a multi-stage recommender system ensemble pipeline with Merlin Systems operators.

    • Perform inference with the Triton Inference Server using the Merlin Systems library.

Running the Example Notebooks#

Containers with the Merlin libraries are available from the NVIDIA NGC catalog. To run the sample notebooks, use the merlin-tensorflow container.

You can pull and run the container.

In production, instead of using the nightly tag, specify a release tag. You can find the release tags and more information on the Merlin TensorFlow container page.

To run the example notebooks using a container, do the following:

  1. After you pull the container, launch it by running the following command:

    docker run -it --gpus all -p 8000:8000 -p 8001:8001 -p 8002:8002 -p 8888:8888 \
      -v <path to your data>:/workspace/data/ --ipc=host \ /bin/bash

    You can remove the --gpus all flag to run the example on CPU.

    The container opens a shell when the run command execution is complete. Your shell prompt should look similar to the following example:

  2. Start JupyterLab by running the following command:

    jupyter-lab --allow-root --ip='' --NotebookApp.token='<password>'

    View the messages in your terminal to identify the URL for JupyterLab. The messages in your terminal should show lines like the following example:

    Or copy and paste one of these URLs:
  3. Open a browser and use the URL provided in the messages from JupyterLab.

  4. After you log in to JupyterLab, navigate to the /Merlin/examples/Building-and-deploying-multi-stage-RecSys/ directory and execute the example notebooks.