Thanks @Louis_Frolio 

Just wanted to clear my use case, I want to run the code locally, but the data insertion should happen in a remote Databricks workspace. I tried using JDBC, but it seems its performance is low in the case of write operations, even after adding batch size and number of partitions.

Is there any alternative for my use case? Also, I am using Java for the current implementation.