Marginal Effects at the Mean/Average Marginal Effects on PySpark

Question

I am currently building a logit regression model on Apache Spark using PySpark (no choice over platform). I have used the GeneralizedLinearRegression model, where the coefficien