独角兽

多元化芯片训练推理引擎研发工程师

阶跃星辰

上海人工智能社会招聘

岗位描述

参与前沿算法的开发、训练与推理,从算法和工程联合调优的角度解决实际算法工程问题;参与国产等多元芯片的导入和适配开发,解决多元芯片与算法适配的工程问题;与公司算法团队深度合作,开展多元化芯片平台前瞻性技术调研并进行自主创新。要求熟练使用 Python 和 PyTorch 训练框架,熟悉 Megatron、vLLM/sglang/comfy、FSDP 等实现原理者优先;能够跟随前沿算法方案并独立完成算法实现;有多模态等跨领域算法经验或国产芯片训练、推理引擎框架开发经验者优先;具备独立解决问题能力和良好团队合作精神。