Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and can work with PyTorch.
conda install nvidia::transformers4rec