【2027】AI框架研发工程师-成都(J14464)

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

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.具备大模型优化经验者优先