jaxdem.forces.law_combiner#

Composite force model that sums multiple force laws.

Classes

LawCombiner([laws])

Sum a tuple of elementary force laws.
class jaxdem.forces.law_combiner.LawCombiner(laws: Tuple[ForceModel, ...] = ())[source]#

Bases: ForceModel

Sum a tuple of elementary force laws.

static force(i: int, j: int, pos: jax.Array, state: State, system: System) → Tuple[jax.Array, jax.Array][source][source]#
static energy(i: int, j: int, pos: jax.Array, state: State, system: System) → jax.Array[source][source]#