Comment by Palmik

I migrated to using the # %% syntax in plain .py files.

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.

https://zed.dev/docs/repl#cell-mode