Pytorch introduces TorchRec, an open source library to build recommendation systems
CEO Mark Zuckerberg has introduced TorchRec, an open source library for building state-of-the-art recommendation Systems under PyTorch, at Inside the Lab event. The new library provides common sparsity and parallelism primitives, enabling researchers to build state-of-the-art personalisation models and deploy them in production. It  includes a scalable low-level modelling foundation alongside rich batteries-included modules. Meta initially targeted“two-tower” architectures with separate submodules to learn representations of candidate items and the query or context. Input signals can be a mix of floating point “dense” features or high-cardinality categorical “sparse” features that require large embedding tables to be trained. Efficient training of such architectures involves combining data parallelism that replicates the “dense” part of computation and model parallelism that partitions large embedding tables across many nodes. The library includes optimised Recommendation Systems kernels that run on FBGEMM, a high-performance kernel library, modelling primitives such as jagged tensors and embedding bags, to create multinodal models using model parallelism. The PyTorch team has released TorchRec after close to two years in testing. TorchRec was used to train a model with 1.25 million parameters that went into production in January and a 3 trillion parameter model that is expected to go into production. Cautiously excited about @PyTorch launching TorchRec. This might be just what we need!— Nirbaay Tandon (@nirbaaytandon) February 23, 2022 Excited to see this https://t.co/eqE11ydI5H from @PyTorch— Josh Patterson (@datametrician) February 23, 2022
TorchRec was used to train a model with 1.25 million parameters that went into production in January.
TorchRec was used to train a model with 1.25 million parameters that went into production in January.
["Open Source AI"]
["Replicate", "Go", "programming_languages:R", "PyTorch", "AI", "recommendation systems", "Scala", "Open Source AI", "ai_frameworks:PyTorch", "programming_languages:Scala", "R"]
["AI", "PyTorch", "recommendation systems", "R", "Go", "Scala", "Replicate", "ai_frameworks:PyTorch", "programming_languages:R", "programming_languages:Scala"]
Poulomi is a Technology Journalist with Analytics India Magazine. Her fascination with tech and eagerness to dive into new areas led her to the dynamic world of AI and data analytics.