机器学习工程师 - 机器学习平台技术

圣弗朗西斯科 2天前全职 网络
2K - 3.1K / 年
总结 我们是一群支持在苹果公司进行基础模型训练的工程师!我们构建基础设施以支持具有通用能力的基础模型训练,如文本、图像、语音、视频和其他模态的理解和生成,并将这些模型应用于苹果产品。我们正在寻找对构建推动深度学习在规模、效率和灵活性方面的系统充满热情的工程师,并为苹果产品的数百万用户带来愉悦体验。 描述 我们正在寻找一名机器学习工程师加入我们的机器学习计算团队,以帮助提高云中模型训练和推理工作负载的效率、可扩展性和可靠性。在这个角色中,您将与高级机器学习工程师、基础设施工程师和研究人员紧密合作,将机器学习工作负载与云基础设施集成,调优性能,并确保加速器的有效利用。 职责 • 支持将模型训练和推理工作负载与基于加速器的云基础设施集成。 • 协助优化机器学习工作负载的性能,以提高吞吐量、延迟和硬件利用率。 • 帮助识别和调试数据加载、模型执行和分布式训练/推理中的性能瓶颈。 • 与高级工程师合作,对模型和基础设施配置进行基准测试。 • 为提高机器学习工作负载的可观察性、可靠性和效率贡献工具、脚本或管道。 最低资格 • 计算机科学、工程或相关领域的学士学位。 • 对机器学习工作流(训练、评估、推理)有基本了解。 • 熟悉Python和至少一个机器学习框架(如PyTorch、TensorFlow、JAX)。 • 具备云计算概念的基本知识(如虚拟机、容器、存储、网络)。 • 对性能优化、系统效率和可扩展的机器学习基础设施感兴趣。 • 具备较强的问题解决能力和学习复杂系统的意愿。 优先资格 • 接触过GPU/TPU计算或基于加速器的工作负载。 • 熟悉分布式训练或推理概念(如数据并行、模型并行)。 • 具有容器化或编排工具(如Docker、Kubernetes)的经验。 • 对机器学习工作负载的分析或基准测试工具有基本了解。 • 参与过与系统、云基础设施或性能工程相关的课程或项目。 薪酬与福利 在苹果公司,基本工资是我们总薪酬包的一部分,并在一定范围内确定。这为您在角色中成长和发展提供了进步的机会。该角色的基本工资范围在$147,400到$220,900之间,您的基本工资将取决于您的技能、资格、经验和地点。 苹果员工还可以通过参与苹果的自由裁量员工股票计划成为苹果股东。苹果员工有资格获得自由裁量限制性股票单位奖励,并可以在自愿参与苹果员工股票购买计划时以折扣价购买苹果股票。您还将获得包括:全面的医疗和牙科保险、退休福利、一系列折扣产品和免费服务,以及与在苹果公司推进您的职业生涯相关的正式教育的某些教育费用(包括学费)的报销。此外,该角色可能有资格获得自由裁量奖金或佣金支付以及搬迁。了解更多关于苹果福利的信息。 注意:苹果的福利、薪酬和员工股票计划需符合资格要求和适用计划或项目的其他条款。 EEO声明 苹果公司是一个致力于包容性和多样性的平等机会雇主。我们致力于在不考虑种族、肤色、宗教、性别、性取向、性别认同、国籍、残疾、退伍军人身份或其他法律保护特征的情况下,为所有申请者提供平等机会。了解更多关于您作为申请者的EEO权利。 苹果公司持续接受对此职位的申请。