An issue reported from ExaSGD, [see here](https://gitlab.pnnl.gov/exasgd/frameworks/exago/-/issues/469#note_63877) @cameronrutherford @abhyshr @cnpetra