Se você usa jupyter diria até que seria mais simples usar algo como o Pycharm.

Mas, entrando na questão de editor o seu problema pode ser em algum plugin ou configuração do vscode, e como tu mencionou que queria outro editor compactivel com as extensões existe o vscodium.

Novamente acredito que seu problema seja mais configuração ou algum plugin do que qualquer outra coisa. Eu te recomendaria remover todos os plugins e ir colocando um de cada vez e ver se o probleam persiste.

Poderia lhe recomendar uma dúzia de editores, mas acredito que não seja isso realmente seu problema.

Espero ter ajudado.