机器学习训练框架研发工程师 - Data AML

团队介绍：Data AML是字节跳动的机器学习中台，为抖音/今日头条/西瓜视频等业务提供推荐/广告/CV/语音/NLP的训练和推理系统。为公司内业务部门提供强大的机器学习算力，并在这些业务的问题上研究一些具有通用性和创新性的算法。同时，也通过火山引擎将一些机器学习/推荐系统的核心能力提供给外部企业客户。 1、负责研发支持字节跳动内部多个日活十亿级应用的推荐广告模型的分布式训练框架； 2、探索世界领先的分布式训练框架新范式，解决如何同时支持超大规模的稀疏模型和稠密模型的难题； 3、深度优化和利用国产新硬件和国际化新型硬件，做到硬件性能的机制优化； 4、深度优化分布式框架能力，打造支持万卡规模训练能力的分布式训练框架； 5、与全公司算法部门紧密合作，为探索下一代推荐广告模型训练新范式做前沿探索和深度优化。 【任职要求】 1、2027届获得本科及以上学历，计算机、软件工程等相关专业优先； 2、熟练掌握Linux环境下的C/C++与Python语言，有良好的编程习惯及编码能力； 3、接触过至少一种机器学习框架（TensorFlow/PyTorch/Jax或其他自研框架）优先； 4、有以下至少一项的背景知识与经验的优先：GPU编程，编译器，高性能网络，分布式存储，集群调度； 5、具有独立解决问题的能力、良好的团队合作精神，有强烈的工作责任心，较好的学习能力、沟通能力和自驱力。