Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch. - NVIDIA-Merlin/Transformers4Rec