You can also run straight python code on Databricks clusters, but it will not run in a distributed manner.

Databricks has a Runtime for Machine Learning that comes with a lot of libraries/frameworks pre-installed. This allows you to run for example PyTorch / TensorFlow code without worrying about infrastructure setup, configuration and dependency management.

Databricks also supports single node clusters to run single-node machine learning workloads