飞腾信息技术有限公司
AI框架适配优化研发工程师
RESPONSIBILITIES
岗位职责
基于计算设备编程框架，开发与主流深度学习框架兼容的算子；
基于计算设备底层软件栈，对主流深度学习框架进行适配开发；
分析主流深度学习框架在计算设备上的性能瓶颈，并进行性能优化；
进行主流大模型并行训练框架的适配与优化。
REQUIREMENTS
任职要求
熟练掌握深度学习算法，能熟练地训练神经网络模型；
熟悉主流框架中的常用算子及其数学意义，有实现经验者优先；
熟悉主流深度学习框架，包括但不限于 PyTorch、TensorFlow、JAX、PaddlePaddle、MindSpore 等，对其中一种或多种的底层实现有较深入研究者优先；
熟悉并行训练框架，了解大模型训练或微调技术，有实际项目经验者优先；
熟悉 CUDA、OpenCL 等异构编程框架，或熟悉 CUDA、oneAPI、ROCm 软件栈者优先；
熟练掌握 Python 和 C/C++，了解 Python 与 C/C++ 相互调用方法；熟悉 Linux，熟练掌握 Shell。 薪资： 面议 学历要求（及以上）： 本科 工作经验： 5-8年 年龄要求： 不限 其他要求： 无
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理，最终申请以企业官方页面为准。