机器学习框架优化工程师

职位描述

1. 完成深度学习算子开发，加速自动驾驶模型的训练和推理，确保模型的高效迭代和实时推理。

2. 确保自动驾驶模型在各类车载平台的工程落地。

任职要求

1. 计算机、软件工程、自动化或相关专业本科及以上学历

2. 熟练使用 C++，熟悉常用算法、数据结构及常见设计模式，具备 1 年以上实际工程经验

3. 熟练掌握 PyTorch 训练和推理加速的方法

4. 熟悉 TensorRT、CUDNN、CUBLAS、CUSOLVER 等计算库，了解 DLA 的使用

5. 掌握 CUDA 编程并具备 1 年以上实际模型加速经验

6. 良好的编程习惯，对自身代码质量的严格要求，良好的工作文档习惯

7. 责任心强，具备优秀的学习能力、独立分析能力和良好的团队沟通能力

加分项：

1. 对机器学习算法模型有深入理解者优先

2. 计算机领域相关的编程大赛获奖者优先

3. AI 算法或系统相关论文发表者优先