Training off of per-atom tensors

[Batbodoo]

Hi! I was interested in training on per-atom tensors (e.g., spins). Is there a current NEP implementation for training off of per-atom tensors? Thanks!

[brucefan1983]

Perhaps you are after this:

https://gpumd.org/dev/nep/input_parameters/atomic_v.html