I find VS-code running notebooks with Colab plugin installed pretty helpful for this type of thing already. (1) run notebook in VS-code https://code.visualstudio.com/docs/datascience/jupyter-noteb... (2) install Github Copilot extension https://marketplace.visualstudio.com/items?itemName=GitHub.c... and then it can quite often achieve what you want with just a comment or some previous code snippets. Then if you load…
I think Colab doesn't support this, but Paperspace Gradient for example does.
https://docs.paperspace.com/gradient/notebooks/notebooks-rem...