Shypd.aiThe largest open index of AI toolsTorch-RecHub is an open-source PyTorch framework for recommendation models, offering a lightweight, efficient, and easy-to-use solution. It provides 30+ mainstream models out-of-the-box and supports one-click ONNX deployment for production-grade systems.
Trending
Also listed in
About
Torch-RecHub is a comprehensive PyTorch framework designed for building and deploying recommendation systems with ease and efficiency. It features a modular design, allowing for easy integration of new models, datasets, and evaluation metrics. Leveraging PyTorch's capabilities, it supports GPU acceleration and Huawei Ascend NPU. The framework boasts a rich library of over 30 classic and cutting-edge recommendation algorithms, including Matching, Ranking, Multi-task, and Generative Recommendation models. It provides standardized pipelines for data loading, training, and evaluation, along with easy configuration via config files or command-line arguments. Reproducibility is a core design principle, and trained models can be exported to ONNX format for seamless production deployment. Additionally, it supports cross-engine data processing with PySpark and offers built-in integration for experiment visualization and tracking with WandB, SwanLab, and TensorBoardX.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
