google colab нельзя. Jupyter Notebook , пожалуйста. Хоть с Pycharm хоть c VSCode. Google colab и Jupyter Notebook это совершенно разные вещи. Google не даст тебе использовать свои вычислительные мощности не в своем приложении.
Не пользовался colab, не не вижу ограничений, думаю можно допилить напильником этот https://github.com/vksmgr/glab/tree/master пример, сделать из него например плагин для идеи и пользоваться