AI驱动工程师

【岗位介绍】

1. AI软件栈的开发调试优化，主要针对于AI Runtime以及Driver层

2. CUDA/HIP生态适配兼容

3. AI框架（如PyTorch/vLLM等）在自研GPU上的适配与部署

4. 客户方案定制与定制，支持AI业务落地

【任职要求】

1. 精通 C++/Python，编程能力优秀

2. 熟练掌握CUDA或HIP编程，了解CUDA/ROCm生态，相关Runtime API以及相关高性能计算库接口

3. 熟悉GPU架构，能结合硬件特性优化

4. 熟悉主流深度学习模型及其流程，精通一种以上主流AI框架（PyTorch/TensorFlow/vLLM/SGLang等）优先

5. 熟悉NCCL/RCCL集合通信库优先

6. 计算机、电子信息、通信工程或相关专业

学历要求：

硕士及以上