AI框架适配优化研发工程师

招聘人数：5
工作地点：长沙、成都
发布日期：2026年9月20日

岗位职责：
1. 基于计算设备编程框架，开发与主流深度学习框架兼容的算子；
2. 基于计算设备底层软件栈，对主流深度学习框架进行适配开发；
3. 分析主流深度学习框架在计算设备上的性能瓶颈，并进行性能优化；
4. 进行主流大模型并行训练框架的适配与优化。

任职条件：
1. 熟练掌握深度学习算法，能熟练地训练神经网络模型；
2. 熟悉主流框架中的常用算子及其数学意义，有实现经验者优先；
3. 熟悉主流深度学习框架，包括但不限于 PyTorch、TensorFlow、JAX、PaddlePaddle、MindSpore 等，对其中一种或多种的底层实现有较深入研究者优先；
4. 熟悉并行训练框架，了解大模型训练或微调技术，有实际项目经验者优先；
5. 熟悉 CUDA、OpenCL 等异构编程框架，或熟悉 CUDA、oneAPI、ROCm 软件栈者优先；
6. 熟练掌握 Python 和 C/C++，了解 Python 与 C/C++ 相互调用方法；熟悉 Linux，熟练掌握 Shell。

薪资：
面议

学历要求（及以上）：
本科

工作经验：
5-8年

年龄要求：
不限

其他要求：
无