工作选择 回国去滴滴做ML 还是留美小公司做普通SDE

拿了国内滴滴的校招offer，做机器学习算法相关的东西， 工资一年33-35万左右。
现在在美国找工作，找到了一家芝加哥当地小金融公司普通SWE的工作，工资一年大概8-9W美元左右。

现在来说就比较纠结， 主要因为在美国工作的话可能就做不了ML，感觉ML可能是未来几年的一个大趋势。而且我个人计划，几年之后还是想回国工作的。所以现在想问问大家的意见~