岗位描述
【工作职责】
1. 针对NVIDIA GPU平台,负责辅助驾驶AI算法的深度性能优化与高效部署,确保实现推理性能;
2. 深入分析TensorRT等NVIDIA推理引擎的性能瓶颈,结合高性能计算库,设计并实施针对性优化方案;
3. 设计开发针AI推理框架及工具链,支持从模型量化、算子优化到异构调度的全链路优化;
4. 与算法团队紧密合作,改进模型结构设计,提升模型在NVIDIA硬件上的执行效率;
5. 构建标准化的性能评估体系,持续优化AI部署流程,推动辅助驾驶功能的量产落地。
【任职要求】
1. 计算机科学、电子信息、自动化等相关专业本科及以上学历,具备扎实的计算机体系结构知识;
2. 至少5年以上AI模型优化经验,精通CUDA编程及性能调优;
3. 熟练掌握TensorRT及其相关生态工具,能够深入理解并优化其内部机制;
4. 对cuBLAS、CUTLASS等高性能计算库有实际使用经验,熟悉GPU底层架构及内存管理;
5. 精通C++/Python编程语言,具备优秀的数据结构与算法基础,熟悉软件工程开发流程;
6. 具备良好的沟通能力,能够与跨职能团队高效协作。