feat(torchsim): add variants support for model outputs#1

HaoZeke wants to merge 1 commit intoCompRhys:mainCompRhys/metatomic:mainfrom

HaoZeke:feat/torchsim-variantsHaoZeke/metatomic:feat/torchsim-variantsCopy head branch name to clipboard