Cleaned up LossAugmentingGroundKernel.