NVIDIA/DeepRecommender
Deep autoencoder implementation for collaborative filtering-based recommendation systems.

Velocity · 7d
+0.5
★ / day
Trend
→steady
star history
This repository implements deep autoencoders for collaborative filtering, as described in a research paper. The model is built with PyTorch and designed to run on GPU with CUDA support. It includes mixed precision training capabilities using NVIDIA Tensor Cores on Volta GPUs. The project demonstrates training on the Netflix Prize dataset for recommendation tasks.