See https://github.com/pytorch/ao/pull/1652/files/21cef83acf7cc358fb0381a78b247cd36a554798#r1962571762