独角兽Momenta
高级算法工程师
北京 / 上海 / 广州 / 苏州/人工智能/社会招聘
负责智能驾驶模型训练优化的研究与落地,通过数据并行、模型并行、通信优化、CUDA 算子优化等手段提升模型训练速度与效率;解决智能驾驶模型训练和部署中的工程与性能问题;负责训练相关工具链和框架的开发、维护与优化。要求具备深度学习算法基础,熟悉 CNN、Transformers 等主流模型原理,具有检测、分类、推荐、自动驾驶等相关项目经验,熟练掌握 C/C++、Python、CUDA 等语言,熟悉 PyTorch 及主流并行训练框架。