【2027】研发工程师（AI方向）-西安(J14228)

岗位职责

1. 01

   参与 GEMM、Attention、MoE、量化/反量化等核心算子的性能分析与优化。

2. 02

   分析 AllReduce、AllGather、All-to-All 等分布式通信开销。

3. 03

   参与 kernel fusion、算子调度、编译优化和硬件特性适配。

4. 04

   使用 profiler 工具定位 GPU kernel、显存访问、通信链路等性能瓶颈。

5. 05

   配合框架和部署团队完成端到端性能优化验证。

任职资格

1. 01

   计算机、电子信息、自动化、数学等相关专业，硕士优先。

2. 02

   熟悉 C/C++ 或 Python，具备扎实的数据结构、操作系统和计算机体系结构基础。

3. 03

   了解 CUDA、GPU 编程、并行计算或高性能计算基础。

4. 04

   理解矩阵计算、Attention、MoE、量化等大模型核心计算模式。

5. 05

   对底层性能优化有强兴趣，具备较好的代码阅读和问题钻研能力。

加分项

1. 01

   有 CUDA、Triton、TVM、MLIR、NCCL、CUTLASS 等实践经验。

2. 02

   参加过 HPC、编译器、并行计算、深度学习系统相关项目。

3. 03

   熟悉 Nsight、CUPTI、torch profiler 等性能分析工具。