@ptrblck As a matter of interest, are there any operations that AREN’T tracked by Autocrat that we should keep an eye out for? I work under the assumption that ANY function/processing bundled in pytorch is tracked - is that a reasonable assumption?