torchref.base.targets.triton.angle module

Triton forward + backward for the bond-angle Gaussian-NLL target.

torchref.base.targets.triton.angle.angle_math_triton(xyz, idx, references_rad, sigmas_rad)

Triton-backed bond-angle Gaussian NLL with analytic backward.

Drop-in replacement for torchref.base.targets.angle.angle_math().