机器学习平台工程师
岗位摘要
构建、分析和优化我们的训练和推理框架;与机器学习团队合作,加速他们的研究和开发,使他们能够开发下一代模型和数据管理工具;研究和集成最先进的技术,以优化我们的机器学习系统;对系统优化有强烈的热情
岗位职责
- 构建、分析和优化我们的训练和推理框架
- 与机器学习团队合作,加速他们的研究和开发,使他们能够开发下一代模型和数据管理工具
- 研究和集成最先进的技术,以优化我们的机器学习系统
任职要求
- 对系统优化有强烈的热情
- 具有多节点大型语言模型训练和推理经验
- 具有开发大规模分布式机器学习系统的经验
- 具备扎实的软件工程技能,熟练掌握CUDA、Pytorch、transformers、flash attention等框架和工具
- 具备出色的书面和口头沟通能力,能够在跨职能团队环境中工作
福利待遇
- 全面的健康、牙科和视力保险
- 学习和发展津贴
- 丰厚的带薪休假
- 可能符合通勤津贴等额外福利的资格