Function gravity_coupling 

pub fn gravity_coupling(
    embedding_a: &[f32],
    embedding_b: &[f32],
    k_base: f32,
    gravity_scale: f32,
) -> f32

Compute gravity coupling between two atoms using embedding similarity. K_ij = K_base × gravity_scale × max(0, cos(sim(e_i, e_j)))