深度学习框架研发工程师-上海

1. 负责智能驾驶模型训练优化的研究和落地,通过数据并行,模型并行,通信优化,CUDA算子优化等手段,大幅提升模型的训练速度与效率。

2. 解决智能驾驶模型训练和部署工作中遇到的各种工程和性能问题。负责训练相关的工具链和框架的开发,维护,和优化。

岗位要求

1. 工作2年以上,计算机/数学/物理/电子工程/自动控制专业硕士及以上学历。思维敏捷,学习能力强。

2. 有扎实的深度学习算法基础,熟悉CNN/Transformers等主流模型原理;在检测,分类,推荐,自动驾驶等深度学习算法领域有不错的项目经验。

3. 优秀的代码能力,熟练掌握C/C++或Python/CUDA等编程语言。有丰富的代码开发经验和良好的工程规范。

4. 能够熟练使用至少一种主流的机器学习框架pytorch,熟悉各种模型/数据并行训练框架优先(deepspeed/megatron);有深度学习训练框架开发经验的加分。

5. 具备扎实的计算机科学功底和编程能力,熟悉计算机体系结构,有高性能计算/高并发开发/系统性能优化经验者优先。具有AutoML/模型推理/训练优化或AI 相关大模型/重要项目经验者优先;

6. 良好的团队协作能力;创新性强,有良好的动手实现能力,对技术有热情。

公司地点:上海浦东新区Momenta(金海园)浦东新区金海园2号楼

公司简介:

Momenta 是全球领先的自动驾驶公司,致力于通过突破性的AI科技,创造更美好的生活。为实现可规模化的自动驾驶,Momenta基于数据驱动的“飞轮”技术洞察,以及量产自动驾驶(Mpilot)与完全无人驾驶(MSD)相结合的“两条腿”产品战略,提供不同级别的自动驾驶解决方案,更高效快速地实现无人驾驶规模化落地,赋能更安全、便捷、高效的未来智慧出行。

Momenta成立于2016年,在中国、德国、日本有全球业务,并已获得上汽集团、通用汽车、梅赛德斯-奔驰、丰田和博世的战略投资。

职位发布者:汪女士

北京魔门塔科技有限公司

融资阶段:

公司规模:100~499人

相似职位: