岗位职责
- 参与AI编译器相关项目的技术方案规划,设计在离线全系统的计算图优化链路;
- 针对搜索/推荐/广告场景,优化模型训练/推理的计算图执行效率;
- 与全公司算法部门深度合作,为重点项目进行算法与系统的联合优化。
职位要求
- 熟悉开源ML编译器代码(如XLA/MLIR/TVM),并有相关开发经验;
- 具有CPU/GPU下的算子Kernel开发和性能优化经验;
- 具有扎实的C/C++编程能力和数据结构和算法知识;
- 具有独立解决问题的能力,能够对业务逻辑进行合理的抽象和拆分,良好的团队合作精神。 加分项:
- 有软硬件联合设计的经验;
- 深入研究过至少一种机器学习框架(TensorFlow/PyTorch/MXNet或其他自研框架)的底层架构和机制。