torchrec
torchrec copied to clipboard
Decouple Table batching + optimizer fusion (module/op level) w/ Sharding
Publicly linking our project proposal for API decomposition, see https://docs.google.com/document/d/1T8oZ8zKbgHuy7q0rkpiyJC5fRUiRugR7oWk799qgFjk/edit#
Feel free to comment directly in the doc.
cc @YLGH @divchenko @zhuzilin