search

facebookresearch / generative-recommenders

Repository hosting code used to reproduce results in "Actions Speak Louder than Words: Trillion-Parameter Sequential Transducers for Generative Recommendations" (https://arxiv.org/abs/2402.17152).
Apache License 2.0
745 stars 141 forks source link

Integration with TorchRec #91

Closed sbhavani closed 4 weeks ago

sbhavani commented 1 month ago

Are there any plans to integrate the embedding_modules or custom samplers back into TorchRec?

jiaqizhai commented 1 month ago

Hi, it should be easy to implement an EmbeddingModule (currently in https://github.com/facebookresearch/generative-recommenders/blob/main/generative_recommenders/modeling/sequential/embedding_modules.py) with EmbeddingCollections from TorchRec. Note you likely want to swap out trainer code as TorchRec uses DMP etc. if you go with this route, so we do not have an official ETA yet, but it should be easy to do so in your own codebase if you use torchrec.