guidance on launching xmanager from colab

The current codelab gives a good overview of orchestrating xmanager using its Python API directly from jupyter cells:
https://github.com/deepmind/xmanager/blob/main/codelab.ipynb

I'm curious if you have any guidance on how to achieve similar result in https://colab.research.google.com/ given it's limited support for `async`, `await` https://github.com/googlecolab/colabtools/issues/1553.