torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Make PositionWeightedModuleCollection faster with torch.export

Open PaulZhang12 opened this issue 8 months ago • 1 comments

Differential Revision: D58830349

PaulZhang12 avatar Jun 20 '24 19:06 PaulZhang12