jax_dna.energy.na1.nucleotide

Extends jax_md.rigid_body.RigidBody for NA1 nucleotide.

Classes

Nucleotide rigid body with additional sites for NA1.

class jax_dna.energy.na1.nucleotide.HybridNucleotide[source]

Nucleotide rigid body with additional sites for NA1.

This class is inteneded to be used as a dataclass for a nucleotide rigid body as a rigid_body_transform_fn in jax_md.energy.ComposedEnergyFunction.

static from_rigid_body(rigid_body: jax_md.rigid_body.RigidBody, dna_com_to_backbone_x: jax_dna.utils.types.Scalar, dna_com_to_backbone_y: jax_dna.utils.types.Scalar, dna_com_to_backbone_dna1: jax_dna.utils.types.Scalar, dna_com_to_hb: jax_dna.utils.types.Scalar, dna_com_to_stacking: jax_dna.utils.types.Scalar, rna_com_to_backbone_x: jax_dna.utils.types.Scalar, rna_com_to_backbone_y: jax_dna.utils.types.Scalar, rna_com_to_stacking: jax_dna.utils.types.Scalar, rna_com_to_hb: jax_dna.utils.types.Scalar, rna_p3_x: jax_dna.utils.types.Scalar, rna_p3_y: jax_dna.utils.types.Scalar, rna_p3_z: jax_dna.utils.types.Scalar, rna_p5_x: jax_dna.utils.types.Scalar, rna_p5_y: jax_dna.utils.types.Scalar, rna_p5_z: jax_dna.utils.types.Scalar, rna_pos_stack_3_a1: jax_dna.utils.types.Scalar, rna_pos_stack_3_a2: jax_dna.utils.types.Scalar, rna_pos_stack_5_a1: jax_dna.utils.types.Scalar, rna_pos_stack_5_a2: jax_dna.utils.types.Scalar) → HybridNucleotide[source]: Class method to precompute nucleotide sites from a rigid body.