it is just a Databricks subscription. When you have subscribed, you can use a GPU based cluster for your ML. There should be a pay-as-you-go version of Databricks, but there is no separate offering for GPU based clusters, it is included.