In case anyone is interested, i created a google colab runtime image based on the google colab runtime that installs latest ROCm and pytorch versions and allows to run google colab notebooks on a ROCm compatible AMD GPU.
I am happily using it with some AI courses that are utilizing Google Colab. I can run the notebooks on the Framework Desktop and GPU is utilized.