Uncertainty-aware Deep Learning with SNGP  |  TensorFlow Core