torchref.base.targets.triton.xray_gaussian module

Triton kernels for the Gaussian X-ray target.

torchref.base.targets.triton.xray_gaussian.gaussian_xray_loss_math_triton(F_obs, F_calc, sigma, mask)

Triton-backed Gaussian NLL X-ray loss.