Is there any solution that we can run AF2 by using GCP TPU vm?

[edwardlo12]

We are trying to infer a sequence with 5000 amino acids. But AF2 can only support single GPU, so we think we can try to use TPU.