jax_md_mod.custom_quantity.kinetic_energy_wrapper# kinetic_energy_wrapper(state, **unused_kwargs)[source]# Wrapper around kinetic_energy to allow kinetic energy computation via traj_util.quantity_traj.