mythos.energy

The energy model and function for mythos.

Submodules

Attributes

DEFAULT_DISPLACEMENT

Functions

energy_fn_builder(...)

Build an energy function from a list of energy functions and configurations.

Package Contents

mythos.energy.DEFAULT_DISPLACEMENT
mythos.energy.energy_fn_builder(energy_fns: list[base.BaseEnergyFunction], energy_configs: list[configuration.BaseConfiguration], transform_fn: collections.abc.Callable[[mythos.utils.types.PyTree], mythos.utils.types.PyTree], displacement_fn: collections.abc.Callable[[mythos.utils.types.PyTree], mythos.utils.types.PyTree] = DEFAULT_DISPLACEMENT) collections.abc.Callable[[mythos.utils.types.PyTree], float][source]

Build an energy function from a list of energy functions and configurations.