机器学习研究工程师,机器学习系统
岗位摘要
构建、分析和优化训练与推理框架;与机器学习团队合作,加速其研发工作,并帮助他们开发下一代模型和数据策划能力;研究并集成最先进技术,以优化机器学习系统;对系统优化充满热情;具有多节点大语言模型训练与推理经验
岗位职责
- 构建、分析和优化训练与推理框架
- 与机器学习团队合作,加速其研发工作,并帮助他们开发下一代模型和数据策划能力
- 研究并集成最先进技术,以优化机器学习系统
任职要求
- 对系统优化充满热情
- 具有多节点大语言模型训练与推理经验
- 具有开发大规模分布式机器学习系统的经验
- 具备扎实的软件工程能力,熟练掌握CUDA、PyTorch、transformers、Flash Attention等框架与工具
- 具备优秀的书面和口头沟通能力,能够在跨职能团队环境中工作
- 加分项:在大语言模型后训练方法及/或下一代用例方面具有丰富经验,包括指令微调、RLHF、工具使用、推理、智能体、多模态等
福利待遇
- 提供具有竞争力的薪酬方案,包括基本工资、股权和福利
- 提供全面的健康、牙科和视力保险
- 提供退休福利
- 提供学习与发展津贴
- 提供充足的带薪休假
- 可能提供通勤津贴等额外福利