@Sajid Thavalengal Rahiman​ Yes please, we have also tried that and found it is not supported for ML Runtimes. but other way to run your data science workloads is use standard DBR above 9.1 LTS and enable photon and run your workloads and keep on tuning your worker type and driver type

Photon runtime | Databricks on AWS