datawhalechina/torch-rechub
A lightweight PyTorch framework for building and deploying recommender system models with modular, extensible architecture.

Velocity · 7d
+0.8
★ / day
Trend
→steady
star history
Torch-RecHub is a PyTorch-based framework designed for rapid development of recommendation system models. It provides over 30 pre-built mainstream recommendation models including CTR prediction and generative recommendation approaches. The framework features modular components for models, datasets, and evaluation metrics, supporting one-click ONNX deployment for production serving across CPU, NVIDIA CUDA, and AMD ROCm hardware platforms.