返回岗位列表

技术团队成员

查看原岗位

地点：美国薪资：薪资未公开日期：2026-01-29

岗位摘要

设计和编写高性能、可扩展的训练软件；从基础设施和代码库性能角度改进训练设置；构建和实施工具以加速训练周期并提高训练基础设施的整体效能；在我们的超级计算和数据基础设施上研究、实施和实验各种想法

岗位职责

设计和编写高性能、可扩展的训练软件
从基础设施和代码库性能角度改进训练设置
构建和实施工具以加速训练周期并提高训练基础设施的整体效能
在我们的超级计算和数据基础设施上研究、实施和实验各种想法
向该领域最优秀的研究人员学习并与之合作

任职要求

极强的软件工程技能
精通Python及相关机器学习框架，如JAX、PyTorch和XLA/MLIR
具备分布式训练基础设施（Kubernetes、Slurm）及相关框架（Ray）的经验
具备使用大规模分布式训练策略的经验
拥有大规模训练大型模型的实践经验，并为训练基础设施的工具和/或设置做出过贡献