python vaex groupby с настраиваемой функцией

Есть ли способ применить настраиваемую функцию к группе с помощью функции groupby vaex DataFrameArray?

Я умею:
df_vaex.groupby(['col_x1','col_x2','col_x3','col_x4'], agg=vaex.agg.mean(df_vaex['col_y']))

Но есть ли способ делать панды:
df.groupby(['col_x1','col_x2','col_x3','col_x4']).apply(lambda x: my_own_function(x['col_y']))