Neural network recommendation models in PyTorch - View it on GitHub