Test multi-element models in nequip-framework.#274

CompRhys wants to merge 2 commits intomainTorchSim/torch-sim:mainfrom

nequip-frameworkTorchSim/torch-sim:nequip-frameworkCopy head branch name to clipboard