Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
Open on GitHub