AI编译器与异构计算青年研究员/青年科学家

我们正在寻找一位在 AI 编译器、计算框架和异构芯片优化领域具有丰富经验的资深人才。该职位负责推动 AI 模型在 NVIDIA GPU及国产 AI 芯片上的适配与性能优化，建设软硬件协同技术能力。 主要职责：

1. 01

   负责 AI 模型在 NVIDIA及国产 AI 芯片上的迁移、适配和性能优化。

2. 02

   设计软硬件协同技术架构，打通模型、框架、编译器、算子、运行时和硬件后端。

3. 03

   开展计算图、动态 Shape、算子融合、内存管理、自动调优和代码生成等编译优化工作。

4. 04

   针对大模型训练与推理，优化计算、显存、通信、吞吐和延迟。

5. 05

   跟踪 AI 编译器、异构计算及国产芯片技术发展，推动关键技术攻关和项目落地。

1. 01

   计算机、电子工程、人工智能等相关专业本科及以上学历。 2、5 年以上 AI 框架、编译器、高性能计算或芯片软件栈研发经验。

2. 03

   熟悉 PyTorch、TensorFlow、JAX或MindSpore等框架的内部机制。

3. 04

   熟悉 LLVM、MLIR、XLA、TorchInductor、Triton或TVM中的一种或多种。

4. 05

   深入理解 NVIDIA GPU、CUDA及相关加速库；具备国产 AI 芯片适配经验。

5. 06

   熟悉计算图优化、Kernel优化、动态 Shape、混合精度、量化及内存优化。

6. 07

   具备复杂技术问题分析、跨团队协作和技术方案落地能力。

优先考虑

1. 01

   参与过 AI 编译器、框架、算子库或大模型推理系统核心研发。

2. 02

   具备国产芯片完整适配或 CUDA/Triton Kernel优化经验。

3. 03

   在相关开源社区有贡献，或拥有论文、专利及大型项目落地经验。