Actually, it can already help you do that!! Seriously, try using gpt4, with a prompt like โoptimize my codeโ and then give it your code. It will do wonders. Of course, it isnt that great with databricks specific syntax as it is with raw python or sql, but the databricks team is getting there!