C++开发实习生(机器学习框架方向)
适用专业
计算机
适用年级
高年级
截止时间
2023-06-21
工作地区
北京
公司介绍
岗位职责
1、参与公司机器学习平台相关组件的开发和维护,包括GPU、CPU分布式训练、GPU推理优化等;
2、支持业务算法团队进行模型效果和训练性能优化;
岗位要求
1、本科及以上学历优先,具备扎实的计算机基础;
2、熟悉C、C++、Python编程语言,有CUDA相关编程经验优先;
3、了解TensorFlow、Pytorch、DeepSpeed、Megatron等开源框架,有源码或相关实习经验优先;
4、了解推理优化相关技术,有编译优化相关实践(TensorRT、XLA、TVM等)优先;
5、具备良好的团队沟通协作能力和学习能力,有责任心和执行力。
投递方式