训练推理优化实习生(Mstar)

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