资深模型训练优化工程师-北京/上海/苏州等地

该职位来源于猎聘 岗位职责 1. 负责智能驾驶模型训练优化的研究和落地，通过数据并行，模型并行，通信优化，CUDA算子优化等，大幅提升模型的训练速度与效率。 2. 深入分析训练系统，定位和解决训练系统各个层次的性能瓶颈，不断提升训练效率，打造业界的训练系统，推进算法的高效迭代。 岗位要求 1. 工作2年以上，计算机/数学/物理/电子工程/自动控制专业硕士及以上学历。 2. 了解深度学习算法基本原理，熟悉CNN/Transformers等主流模型原理；具有AutoML/模型推理/训练优化或AI 相关大模型/重要项目经验者优先； 3. 优秀的代码能力，熟练掌握C/C++或Python/CUDA等编程语言。 4. 能够熟练使用至少一种主流的机器学习框架pytorch，熟悉各种模型/数据并行训练框架优先(deepspeed/megatron)；有深度学习训练框架开发经验的加分。 5. 具备扎实的计算机科学功底和编程能力，熟悉计算机体系结构，有高性能计算/高并发开发/系统性能优化经验者优先。 6. 良好的团队协作能力；创新性强，有良好的动手实现能力，对技术有热情。 可以base北京/上海/深圳/苏州等地；