Is there any extension of vim/nvim that enable editing and runing jupyter notebook like the Jupyter in VSCode？

Question

Is there any extension of vim/nvim that enable editing and runing jupyter notebook like the Jupyter in VSCode？

Like in this: enter image description here