← all repositories

meta-pytorch/torchrec

A PyTorch extension library enabling massive-scale recommendation system training across distributed GPUs.

2.6k stars Python Domain AppsML Frameworks
torchrec
Velocity · 7d
+1.4
★ / day
Trend
steady
star history

TorchRec provides common sparsity and parallelism primitives for large-scale recommendation systems. It enables training and inference of models with embedding tables sharded across many GPUs using hybrid data-parallelism and model-parallelism. Key features include automated sharding planners, pipelined training, and optimized kernels via FBGEMM for powering production recommender systems at scale.

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