【2027】AI框架研发工程师(J14435)

岗位职责（满足以下任意一方向）：

1. 负责pytorch/TF/JAX及配套生态框架组件在国产GPGPU上的适配、开发与系统级优化；
2. 基于国产GPGPU进行多领域模型（如搜广推、视觉推理、模型训练等）的端到端性能分析及优化；
3. 负责基于国产GPGPU的兼容能力开发及高性能组件（如向量数据库等）研发；
4. 跟踪AI框架及场景的前言进展，依托国产GPGPU进行实际工程落地；

岗位要求：

1. 计算机科学、电子工程、通信工程、信息科学或相关专业硕士及以上学历
2. 熟悉 Linux 开发环境，熟悉主流人工智能框架及性能优化方法，掌握 C++ / Python 编程，熟悉 CUDA 编程模型及 GPU 执行与内存体系。
3. 熟悉至少一种主流AI框架，或具备相关开发与性能优化经验，包括但不限于pytorch、triton、torchtitan、tensorflow、jax、xla、iree等。
4. 具备 LLM/ VLM/搜广推等任意领域的训练或推理优化经验。

5.加分项：
a.具备CPU、GPU算子优化经验，有并行计算相关背景者优先；
b.具备TVM、MLIR、XLA、Triton等编译优化或图算优化经验者优先
c.具备国产加速卡相关经验者优先
d. 具备异构集群经验者优先
e.具备大模型优化经验者优先