POC Implementation: https://github.com/pytorch/pytorch/pull/142114 RFC: https://docs.google.com/document/d/1r_nsPy01H-mUH-rIVxrHfvn-_ttrt7c5lVYfIrpD1wU/edit?tab=t.0#heading=h.9zgk98h1ce53
