NVIDIA-Merlin/Transformers4Rec
NVIDIA library that applies transformer architectures to sequential and session-based recommendation problems using PyTorch.

Velocity · 7d
+0.7
★ / day
Trend
→steady
star history
Transformers4Rec bridges natural language processing and recommender systems by integrating with Hugging Face Transformers to bring state-of-the-art transformer models to recommendation tasks. It processes sequences of user interactions (browsing, purchases, etc.) to predict the next item a user is likely to interact with. The library is built on PyTorch and supports modern transformer architectures adapted for tabular interaction data.