岗位描述
岗位职责（满足以下任意一方向）： 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.具备大模型优化经验者优先 职位类别: 校园招聘