独角兽

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

阶跃星辰

上海人工智能社会招聘

岗位描述

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