Is it possible to convert this code to python on Spark?

[Mr-Java-Chen-HK]

# Hey all,
i am interested in moving this code to Spark to lower the computing time consumption.(With CUDA,the manager on company still thinks it is too slow).Can anyone tell me is it worthy to do it?Since I am a newer to Spark,if anyone gives me some advice,it will be greatly appreciated! @andersbll @anders @boesen @Lindbo @larsen

[zengqi-ustb]

I also have trouble with this issue.Looking forward to the discussion!Hope someone familiar with Spark would do me a favor!