jax_md_mod.custom_quantity.kinetic_energy_tensor# kinetic_energy_tensor(state)[source]# Computes the kinetic energy tensor of a single snapshot. Parameters: state – Jax_md simulation state Returns: Kinetic energy tensor