The next generation of Jupyter, JupyterLab is turning into a whole IDE/platform, not just a notebook. They are also trying to incorporate the monaco editor which drives VS Code. Wouldn't that be a better solution?
VSCode's magic isn't just the monaco editor, IMO it's really the ecosystem and really nice Intellisense the other extensions provide.
though I do agree, this might be the better solution in the long run.
https://docs.google.com/presentation/d/1n2RlMdmv1p25Xy5thJUh...
It was eye opening.
His YouTube live coding on building a neural net from scratch really shows the power of using VSCode to do AI work and complements his assertions.
https://www.youtube.com/watch?v=o64FV-ez6Gw&feature=youtu.be
Both are well worth the time. All the reading I've been doing on ML/AI really came together after watching and taking notes in that YouTube video.