I spent time reading through Joel Grus' "I hate notebooks" slide deck and he makes a really compelling why they (Jupyter notebooks) are not the right tool for many reasons.
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.