jax_md_mod.custom_energy.tabulated_pair

Contents

jax_md_mod.custom_energy.tabulated_pair#

tabulated_pair(displacement_or_metric, x_vals, y_vals, degree=3, r_onset=0.9, r_cutoff=1.0, species=None, per_particle=False)[source]#

Convenience wrapper to compute tabulated energy over a system.

Return type:

Callable[[Array], Array]