Maybe ComENet is not structure invariant?

[StefanIsSmart]

When we use original xyz and xyz + 10 (just translation)
The output is different.

[StefanIsSmart]

I change the forward process, and pass in the batch, xyz and atom type.
We can find the result is different between to same structures.