Thank you! Sad to know this is an unresolved issue. I’m trying to integrate libtorch as a plugin yet having the same problem and can’t figure how it is solved in the two repo you mentioned. The issue is also present if trying to perform a forward over an nn Module, so I guess it’s related to how either torch or UE deals with memory…