I have come across packages that specialize in fitting energy and forces, but none seem to include stress. I would greatly appreciate it if you could recommend packages that are capable of fitting all three parameters—force, energy, and stress—for neural network interatomic potentials.