fn d_theta_chunk<B: Backend>(
phys_bhlpr: &F<B, 5>,
d_phys_bhlpr: &F<B, 5>,
num_angles_total: usize,
rope_dim: usize,
rotate_pairwise: bool,
) -> F<B, 4>Expand description
The per-pair angle gradient of the de-rotation (see the module header).