Training for stress

[JonathanSchmidt1]

To train force fields for crystal structures e.g. faenet on some of the available datasets would require stress from automatic differentiation.
Is this already available and I missed it or anybody working on it?
best,
Jonathan