Hi @voz, thanks a lot for getting back to me so quickly. I’ve had a quick look, and it is exactly what I was searching for. I’ve got another question regarding the transformation from torch IR to aten IR. Could you provide an example of how to perform these transformations or direct me to the main entry point in the code?