Enable gpu/tpu training with XManager.

google / uncertainty-baselines

High-quality implementations of standard and SOTA methods on a variety of tasks.

Apache License 2.0

1.45k stars 202 forks source link

Enable gpu/tpu training with XManager. #1311

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

Enable gpu/tpu training with XManager.

google-cla[bot] commented 1 year ago

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.