Simulate multiple initial conditions simultaneously?

[jeevster]

Is there a way in the codebase to simulate MD with Nequip on multiple initial conditions in parallel (i.e choose 100 ics and run MD in parallel for each of them, avoiding any explicit loops)? Would be great if this could be done in a way that takes advantage of Pytorch vectorization.