How to use a GROUPED_MAP pandas udf in every Spark Dataframe partition?

Question

I would like to use a pandas UDF to speed up a user defined function. The type of pandas udf I am interested in the one that gets a pandas DataFrame as input and returns a Panda