Multi-GPU Example Notebooks

The following notebooks demonstrate how to perform multi-GPU training with NVTabluar, TensorFlow, and PyTorch:

• Training with TensorFlow on MovieLens Data

• Training with PyTorch on MovieLens Data

• Learn about NVTabular and Dask

See the examples/multi-gpu-movielens of the repository on GitHub to view the following that are related to the notebooks:

• hvd_wrapper.sh

• tf_trainer.py

• torch_trainer.py

• torch_trainer_dist.py