Is there equivalent of scipy distance_matrix in pyspark?

Question

I want to calculate distance matrix for two dataframes a and b In python I could do it with

from scipy.spatial import distance_matrix

distance_matrix(a.values,