Kevin from the Colab team. We just released an open source VS Code extension for connecting notebooks in VS Code to Colab runtimes. Give it a try and let me know what you think - here, or on the GitHub repo's discussions or issues.