I need to run pyspark using gpu. I choosed numba for that task.
I had an attempt trying to use cuda decorator on pyspark aggregation function.