多元化芯片训推引擎研发工程师
岗位摘要
参与前沿算法的开发和训练推理，从算法和工程联合调优的角度解决实际算法工程问题；参与国产等多元芯片的导入和适配开发，解决多元芯片与算法适配的工程问题；与公司算法团队深度合作，进行多元化芯片平台前瞻性技术调研并进行自主创新
岗位职责
- 参与前沿算法的开发和训练推理，从算法和工程联合调优的角度解决实际算法工程问题
- 参与国产等多元芯片的导入和适配开发，解决多元芯片与算法适配的工程问题
- 与公司算法团队深度合作，进行多元化芯片平台前瞻性技术调研并进行自主创新
任职要求
- 熟练使用Python和PyTorch训练框架，熟悉Megatron、vLLM/SGLang/Comfy、FSDP等实现原理者优先
- 跟随前沿算法方案，并能独立完成算法实现
- 拥有多模态等跨领域算法经验者优先
- 拥有国产芯片训练和推理引擎框架开发经验者优先
- 具有独立解决问题的能力，良好的团队合作精神
福利待遇
- 薪资35-60K，16薪
工作地址
北京海淀区大恒科技大厦12F