机器学习工程师,运行时与优化

25天前全职 网络
145.3万 - 184.5万 / 年 Waymo

Waymo

旧金山
Waymo 是一家自动驾驶技术公司,使命是成为全球最值得信赖的司机。自 2009 年作为 Google 自动驾驶汽车项目开始以来,Waymo 专注于构建 Waymo Driver——全球最有经验的司机™——以改善出行的可及性,同时拯救成千上万因交通事故而失去的生命。Waymo Driver 驱动 Waymo 的全自动打车服务,并可应用于多种车辆平台和产品用例。Waymo Driver 已经提供了超过一千万次仅乘客的行程,凭借其在公共道路上自主驾驶超过一亿英里以及在 15 个以上美国州进行数十亿次模拟的经验。 Waymo 的 ML 平台团队提供一套工具,以支持和自动化机器学习工作流的生命周期,包括特征和实验管理、模型开发、优化和监控。这些努力使得机器学习对 Waymo 的团队(包括感知、规划、研究和模拟)更加可及。 我们正在寻找具有机器学习软件或机器学习系统专业知识的工程师,帮助我们改善云端和汽车上的计算性能。您将从系统的角度跨越整个机器学习堆栈工作,从高效的深度学习模型、模型压缩、机器学习软件(例如 JAX、XLA、Triton 和 CUDA)开始。您将面临将 Waymo 机器学习模型部署在有限计算资源上的挑战。在这个混合角色中,您将向运行时与优化的高级经理汇报。 您将: • 与世界一流的 Waymo 机器学习科学家在感知、规划、研究和模拟方面进行合作。识别系统和模型中加速机器学习工作负载的机会。 • 通过培养初级工程师来领导项目,从提案到执行。 • 分析并改善云端和自动驾驶汽车上的机器学习系统工作负载。 • 将模型优化、高效的深度学习技术和机器学习软件改进应用于 Waymo 的机器学习系统。 您拥有: • 计算机科学、电子工程、深度学习或相关领域的硕士学位。 • 2 年以上作为技术负责人工作的经验,包括撰写项目计划、与客户团队互动、指导、负责目标和执行、报告状态。 • 5 年以上在机器学习系统或机器学习软件堆栈(Pytorch/JAX/TF、运行时库、机器学习编译器)中开发解决方案的经验。 • 深刻理解机器学习系统架构、性能分析和工具。 • 较强的 Python 或 C++ 编程能力。 我们希望您拥有以下一项或多项: • 计算机科学、电子工程、深度学习或相关领域的博士学位。 • 熟悉机器学习硬件加速器(例如 GPU/TPU)的硬件架构。 • 对基础模型或大型语言模型的模型优化或高效深度学习技术有深入了解。 • 具备 GPU 硬件或 TPU 硬件及相关系统软件的经验。 该全职职位在美国各地的预期基础薪资范围如下。实际起薪将基于与工作相关的因素,包括确切的工作地点、经验、相关培训和教育以及技能水平。在招聘过程中,您的招聘人员可以分享有关角色地点的具体薪资范围,或者如果角色可以远程执行,则可以分享您首选地点的具体薪资范围。 Waymo 员工还可以参加 Waymo 的酌情年度奖金计划、股权激励计划和慷慨的公司福利计划,具体取决于资格要求。 薪资范围: $204,000—$259,000 美元