Insertion uses all memory

Hi all, 

In my PySpark script, I call ALEX to insert around 1 million records. In some cases, ALEX uses all memory and the program crashes. 
I am still trying to figure out the specific cause. I wonder if anyone has similar issues. Any input is appreciated. 

Bests,