关于分布式机器学习框架的技能点问题

现在研二，研究生的方向是机器学习。研一出去实习过一段时间，发现实际上基本都是数据挖掘，以抽 feature 为主，不太感兴趣。加上本科毕设做的是分布式数据库的东西，想转做分布式机器学习框架，就是去造轮子。正好之前的任务要看 tensorflow 的源码，现在的任务也是在 spark 的基础上做机器学习的库。因为现在找工作还有一年时间，想问下如果想在一年之后找相关方面的工作，现在应该做什么准备？另外国内感觉造轮子的地方不多，有没有什么值得关注的部门或者公司？

PS ：自己也看了一些资料，目前打算先刷完 MIT 的 6.824 ，然后可能自己写一点简单的 graphLab 或者 ps ？但感觉刷完 6.824 就要开始刷题，也没什么时间了。