A Microsoft desenvolveu várias extensões para usar o Jupyter no VS Code.

https://marketplace.visualstudio.com/publishers/Microsoft

Eu uso o Night Wolf (gray com monokai), Night Owl também é boa.

https://marketplace.visualstudio.com/items?itemName=MaoSantaella.night-wolf