岗位描述(节选)
岗位职责:
1.持续开发torch_mlu, 集成PyTorch最新技术,保持torch_mlu技术竞争力和创新性;
2.解决网络推理和训练精度、性能和分布式问题
3.协助分析和解决torch_mlu应用问题
岗位要求:
1. 较强的C++开发经验和一定Python开发能力,扎实的数据结构和算法基础
2. 精通一种主流框架开发- PyTorch/Tensorflow/vllm等
3. 解决过网络精度、性能问题
4. 熟悉分布式训练的原理和调优
5. 学习能力强,有强烈的求知欲和自我驱动能力,良好的沟通和团队协作能力
6. 计算机及相关专业毕业
7. 加分项1:参与过AI框架相关社区开发,提交过代码
8. 加分项2:熟悉torch inductor、Triton、XLA、MLIR、TVM等
9. 加分项3:解决分布式训练问题的经验