I wish Zed would implement support for Jupyter notebooks first. Maybe this sounds like a thing I can contribute
For me, it's a superior experience anyway. I also prefer it in editors that support both (like VS code).
You can run the REPL with a Jupiter kernel as well.