I want to have databricks related developement locally.

There is extension that allows to run local python file on remote databricks cluster.

But I want to have cell like structure that is present in databricks UI for python files in local IDE as well.

Is there any such extension for any IDE

Thanks