feat(torch.compile): partial PET application#1029

HaoZeke wants to merge 1 commit intometatensor:mainmetatensor/metatrain:mainfrom

HaoZeke:petPartialCompileHaoZeke/metatrain:petPartialCompileCopy head branch name to clipboard