Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training
The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX