jax_md_mod.custom_energy.tabulated

Contents

jax_md_mod.custom_energy.tabulated#

tabulated(dr, spline, **unused_kwargs)[source]#

Tabulated radial potential between particles given a spline function.

Parameters:
  • dr (Array) – An ndarray of pairwise distances between particles

  • spline (Callable[[Array], Array]) – A function computing the spline values at a given pairwise distance.

Return type:

Array

Returns:

Array of energies