AI框架研发工程师(J14476)

（满足以下任意一方向）

1. 01

   负责pytorch/TF/JAX及配套生态框架组件在国产GPGPU上的适配、开发与系统级优化；

2. 02

   基于国产GPGPU进行多领域模型（如搜广推、视觉推理、模型训练等）的端到端性能分析及优化；

3. 03

   负责基于国产GPGPU的兼容能力开发及高性能组件（如向量数据库等）研发；

4. 04

   跟踪AI框架及场景的前言进展，依托国产GPGPU进行实际工程落地；

1. 01

   计算机科学、电子工程、通信工程、信息科学或相关专业硕士及以上学历

2. 02

   熟悉 Linux 开发环境，熟悉主流人工智能框架及性能优化方法，掌握 C++ / Python 编程，熟悉 CUDA 编程模型及 GPU 执行与内存体系。

3. 03

   熟悉至少一种主流AI框架，或具备相关开发与性能优化经验，包括但不限于pytorch、triton、torchtitan、tensorflow、jax、xla、iree等。

4. 04

   具备 LLM/ VLM/搜广推等任意领域的训练或推理优化经验。

5. 05

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