jax_md_mod.custom_quantity.temperature

Contents

jax_md_mod.custom_quantity.temperature#

temperature(state, mask=None, **kwargs)[source]#

Temperature function that is consistent with quantity_traj interface.