【职位职责】

1. 根据需求，调研指定的AI算法、模型、训练或推理技术并撰写报告；

2. 对特定算法或技术进行深入分析、推导，编写测试代码，配合IC团队进行

前期仿真验证并优化性能；

3. 编写并优化AI算子的核函数实现。

【职位要求】

1. 硕士及以上学位，电子、通信、计算机、物理、数学或相关专业；

2. AI 行业3年以上工作经验；

3. 熟悉AI算法原理，特别是基于Transformer架构的模型；

4. 较丰富的大模型训练、调优经验；

5. 熟悉PyTorch、TensorFlow、Megatron、DeepSpeed 等训练框架；

6. 熟悉卡间通信及使用，如OpenMPI、nccl；

7. 熟练cuda、HIP、OpenCL三者中至少一种编程；

8. 加分项: 开源LLM代码贡献者；了解GPU、NPU等芯片架构；有大规模

LLM 推理部署经验。