jax_dna.energy
The energy model and function for jax_dna.
Submodules
Attributes
Functions
|
Build an energy function from a list of energy functions and configurations. |
Package Contents
- jax_dna.energy.DEFAULT_DISPLACEMENT
- jax_dna.energy.energy_fn_builder(energy_fns: list[base.BaseEnergyFunction], energy_configs: list[configuration.BaseConfiguration], transform_fn: collections.abc.Callable[[jax_dna.utils.types.PyTree], jax_dna.utils.types.PyTree], displacement_fn: collections.abc.Callable[[jax_dna.utils.types.PyTree], jax_dna.utils.types.PyTree] = DEFAULT_DISPLACEMENT) collections.abc.Callable[[jax_dna.utils.types.PyTree], float][source]
Build an energy function from a list of energy functions and configurations.