TPU support in Colab code: switch to the "TPU v2" runtime on a TPU VM

tensorflow / gnn

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

Apache License 2.0

1.34k stars 171 forks source link

TPU support in Colab code: switch to the "TPU v2" runtime on a TPU VM #796

Closed copybara-service[bot] closed 6 months ago

copybara-service[bot] commented 6 months ago

TPU support in Colab code: switch to the "TPU v2" runtime on a TPU VM and retire workarounds for the older "TPU (deprecated)" runtime that uses a separate TPU Node.

Along the way, changing runtimes upgrades from TF2.12 to TF2.15.

review-notebook-app[bot] commented 6 months ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.