jax_md_mod.custom_quantity.init_bond_angle_distribution

jax_md_mod.custom_quantity.init_bond_angle_distribution#

init_bond_angle_distribution(displacement_fn, bond_angle_params, reference_box=None)[source]#

Initializes a function computing a dihedral distribution.

Parameters:

  • displacement_fn – Displacement to compute dihedral angles

  • bond_angle_params (BondAngleParams) – Struct describing the dihedral angles and expected format of the computed distribution

  • reference_box – Unused

Returns:

Returns a function that computes a distribution of dihedral angles given a simulation state.

Contents