Add temperature scaling to Monte Calro sampling for softmax estimation; fix a bug
Add temperature scaling to Monte Calro sampling for softmax estimation; fix a bug

PiperOrigin-RevId: 436271619

google/uncertainty-baselines