Kumaran
Databricks Employee
Databricks Employee

Hi @ml-engineer Thank you for contacting Databricks community support.

I think you are installing tensorflow version 2.14, could you update to 2.16 using the pip install --upgrade tensorflow and see how it works?