Batchify the stress tensor routines

The following routines can be 'easily' batchified:

• [] stress_from_kinetic_energy_electron
• [] stress_from_pseudo