torchref.base.targets.triton.chiral module

Triton forward + backward for the chiral-volume target.

torchref.base.targets.triton.chiral.chiral_math_triton(xyz, indices, ideal_volumes, sigmas)[source]

Triton-backed chiral-volume Gaussian NLL with analytic backward.