← all repositories

datawhalechina/torch-rechub

A lightweight PyTorch framework for building and deploying recommender system models with modular, extensible architecture.

1.2k stars Jupyter Notebook ML FrameworksDomain Apps
torch-rechub
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.

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