Migrate files from /third_party/py/uncertainty_baselines/experimental/shoshin to /third_party/py/skai. Small fixes to adhere to /third_party/py build requirements.#1293
Open
copybara-service[bot] wants to merge 1 commit intomainfrom test_530732086
-5,293
