Non-linear machine learning models incorporating SNPs and PRS improve polygenic prediction in diverse human populations