← all repositories

NVIDIA-Merlin/Transformers4Rec

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

1.3k stars Python Domain AppsML Frameworks
Transformers4Rec
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.