算子开发工程师

该职位来源于猎聘 算子开发工程师 岗位职责： 1. 利用深度学习编译技术，结合专家级经验（算法/系统/硬件），解决搜广推、CV、NLP，以及多模态、大模型的训练和推理场景挑战。 2. 系统化分析模型，提炼/抽象优化方法，寻求解空间（包括算子调度，融合，Schedule，异构调度等）。 3. multi-framework（例如Tensorflow、Pytorch等） 计算图优化Pass开发，手段包括不限于编译IR优化，专家经验模板化等等。 4. multi-device（例如 GPU，CPU，NPU）专项 SIMT、SIMD 计算加速，包括核心 Gemm 算子，Reduce算子，以及复杂融合算子等等。 岗位基本要求： 有扎实的计算机理论基础，熟练掌握C++和Python语言 具备以下优先： 1. 熟悉PyTorch 或 TensorFlow等引擎的架构和运行原理，有深度定制经验优先 2. 熟悉TVM/MLIR/XLA等DL编译器，有Cutlass技术经验，具备计算图翻译至IR并优化的经验优先 3. 在深度学习领域有丰富的模型优化经验，熟悉GPU或NPU架构，有CUDA或NPU编程经验优先 4. 学习能力和好奇心强，具备良好的问题分析和解决能力，有团队合作精神和沟通能力