大模型Infra工程师 (PRC)

【岗位职责】 1、核心算子开发与极致优化：基于华为昇腾（Ascend 910B等）、平头哥等国产AI加速芯片，进行深度学习核心计算算子（如Attention、Linear、LayerNorm、MoE等）的高性能定制开发与迭代优化； 2、AI框架与系统适配：负责将主流开源AI框架（如PyTorch、MindSpore）及大模型训练/推理工具链（如Megatron-LM、DeepSpeed、vLLM等）深度适配到国产算力平台上，确保功能正确性与系统稳定性； 3、端到端性能调优：使用底层性能分析工具（如Ascend Profiler等）进行算子级与模型级的性能瓶颈分析。针对计算、访存、通信瓶颈制定调优策略（如算子融合、混合精度、计算与通信重叠设计）； 4、编译优化与前瞻探索：参与探索TVM、MLIR、Triton等AI编译器技术在国产异构后端的应用，提升算子自动生成效率与多硬件平台的泛化适配能力…… 完整岗位描述与官方投递入口,请登录后在岗位详情页查看。

本站当前在招 21736 个北京岗位（全城各职能合计），其中 1797 个公开标注了薪资。按这些标注统计：月薪（税前）主流区间 7K ~ 20K（25~75 分位），中位约 12K，最低 1K、最高 300K。