Make MACE export not break the ability to save a checkpoint#1020

Luthaf merged 2 commits intomainmetatensor/metatrain:mainfrom

mace_exportingmetatensor/metatrain:mace_exportingCopy head branch name to clipboard