Torch-RecHub: 軽量で効率的かつ使いやすいPyTorch推薦フレームワーク

Torch-RecHubは、推薦システムを構築するためのPyTorchベースのフレームワークです。30以上の推薦モデル（ランク付け、マッチング、マルチタスク、生成タスク）をサポートするモジュール設計を備えており、データロード、トレーニング、評価の標準化されたパイプラインを提供します。設定は設定ファイルまたはコマンドライン引数で簡単にできます。主な機能には、本番環境デプロイ用のONNXエクスポート、PySparkなどのクロスエンジンデータ処理、WandB、SwanLab、TensorBoardXを用いた組み込み実験追跡が含まれます。CPU、NVIDIA CUDA、AMD ROCm、Huawei Ascend NPUの複数のハードウェアバックエンドをサポートしています。CTR予測（例：CriteoでのDeepFM）、マッチング（例：MovieLensでのDSSM）、マルチタスク学習（例：MMoE）などのタスク用の例題スクリプトが提供されています。プロジェクト構造は、コアコンポーネント（モデル、トレーナー、ユーティリティ）と例題、チュートリアル、テストを分離しています。