Use the "main" branch in links to the Uncertainty Baselines GitHub repo.
Use the "main" branch in links to the Uncertainty Baselines GitHub repo.

PiperOrigin-RevId: 382407670

tensorflow/docs,tensorflow/docs,tensorflow/docs