torchrec
torchrec copied to clipboard
pytorch
Pytorch domain library for recommendation systems
Differential Revision: D62032704
Summary: As titled Differential Revision: D61371128
Hi, thank you for this great work. I would like to know which [distributed strategy](https://github.com/Lightning-AI/pytorch-lightning/blob/76d3d22c5997398ffb5296cf500c723a176c0a06/src/lightning/pytorch/trainer/trainer.py#L95) to use with lightning trainer. I see two potential avenues: 1. DDP strategy: [following this...
Summary: * Added model wrappers for DeepFM and DLRM. The wrapper will take ModelInput as an only parameter in the forward method. * Added the unit tests to cover the...
Hi, I’m trying to run a batch transform job on SageMaker, but I’m encountering an issue: the SageMaker instance uses an older NVIDIA driver that only supports CUDA runtime version...
Summary: Publish change to enable KVEmbeddingInference when use_virtual_table is set to true Differential Revision: D75321284
## PR Summary This small PR resolves the annoying deprecation warnings of the `logger` library: ```python DeprecationWarning: The 'warn' method is deprecated, use 'warning' instead ``` See [CI logs](https://github.com/pytorch/torchrec/actions/runs/15201361250/job/42755956905#step:15:5170) for...
Summary: This commit introduces enhancements to the optimizer configuration in TorchRec. It now supports specifying the optimizer type, learning rate, momentum, and weight decay. These changes provide more flexibility and...
Summary: Adding the planner input context has to the sharding plan name that's uploaded to manifold. We will use this hash in the next diff to validate that the sharding...
