torchrec

torchrec copied to clipboard

mark metricmodule methods as experimental

2

Summary: tsia Reviewed By: kausv Differential Revision: D75014609

iamzainhuda

OOM when I create a 196GB embedding.

2

I want to create a total of 196GB embedding, where each embedding is calculated as (274352000 buckets * 192 dims * 4 bytes * 1), distributed across 4 * 8...

zhaozheng09

roadmap 2025 H1 H2

1

Where can I see a roadmap? like https://drive.google.com/file/d/1daGLhmlWezP4C35PfdQMkGMC4ePEPYor/view

zhaozheng09

Does torchrec support implicit feedback datasets?

1

Does torchrec support implicit feedback datasets where we only have data on the items that the customers purchased? How about where with datasets that have sparse data where >60% null...

JorgeBonillaSD

Enable fx tracing for maybe_td_to_kjt

1

jeetkanjani7

Fix undeclared type in colab for sharding examples

1

Summary: Fixed the undeclared type "ShardingPlan" Differential Revision: D75108254

lizhouyu

Support RecMetrics new options in Pyper config

2

Summary: Fix the issue in tower QPS metric update issue in P1806989639 and discussed in [link](https://fb.workplace.com/groups/527654686243695/posts/989373686738457/?comment_id=990338486641977&reply_comment_id=990584749950684), and expose the new TorchMetric fusion [feature](https://fb.workplace.com/groups/527654686243695/permalink/989373686738457) in to Pyper users Differential Revision: D75120928

HugeEngine

create fx argument for sparsefeature types

2

Summary: * make torchrec feature container types fx argument construction friendly. easily for tracing. Reviewed By: 842974287 Differential Revision: D40275521

YazhiGao

fix state_dict compatibility with ebc family

2

Summary: * add state_dict, load_state_dict overload to unify quant_ebc state_dict with the ebc family. Differential Revision: D40236094

YazhiGao

fix dmp copy

2

Summary: * since dmp's wrapped module interface mismatch. the copy routine based on named_module replacement will fail. thus we directly copy the wrapped one underneath. Differential Revision: D44574214

YazhiGao