jax_dna.energy.na1.fene
FENE energy function for NA1 model.
Classes
Configuration for the FENE energy function. |
|
FENE energy function for NA1 model. |
Module Contents
- class jax_dna.energy.na1.fene.FeneConfiguration[source]
Bases:
jax_dna.energy.configuration.BaseConfigurationConfiguration for the FENE energy function.
- dna_config: jax_dna.energy.dna1.FeneConfiguration | None = None
- rna_config: jax_dna.energy.dna1.FeneConfiguration | None = None
- required_params: tuple[str] = ('nt_type', 'dna_eps_backbone', 'dna_r0_backbone', 'dna_delta_backbone', 'dna_fmax', 'dna_finf',...
- init_params() FeneConfiguration[source]
Initializes the dependent parameters in configuration.
Should be implemented in the subclass if dependent parameters are present.
- class jax_dna.energy.na1.fene.Fene[source]
Bases:
jax_dna.energy.base.BaseEnergyFunctionFENE energy function for NA1 model.
- params: FeneConfiguration
- __call__(body: jax_dna.energy.na1.nucleotide.HybridNucleotide, seq: jax_dna.utils.types.Sequence, bonded_neighbors: jax_dna.utils.types.Arr_Bonded_Neighbors_2, unbonded_neighbors: jax_dna.utils.types.Arr_Unbonded_Neighbors_2) jax_dna.utils.types.Scalar[source]
Calculate the energy of the system.