Stack AV 机器学习训练基础设施高级软件工程师（远程，宾夕法尼亚州）
负责设计可扩展的机器学习训练基础设施和端到端模型流水线。 通过平台工程、性能优化和跨团队协作，助力推进 Stack AV 的自动驾驶卡车系统。
描述
- 构建并改进高性能训练平台组件，涵盖编排、训练抽象、控制平面服务、可观测性和性能优化。
- 开发完整的机器学习流水线，涵盖日志处理、特征提取、数据集架构与存储、模型配置、训练、性能分析和加速。
- 评估训练基础设施，发现并解决性能瓶颈。
- 推动系统抽象和开发者工具，帮助机器学习工程师快速迭代模型。
- 采用开源技术，让机器学习工程师能够独立分析并改进工作流程。
- 坚持高标准的工程实践，帮助营造以技术卓越为核心的团队文化。
- 负责高性能、多租户 AI 训练平台的架构设计与实施。
- 与机器学习平台、基础设施、自动驾驶和安全评估团队开展合作。
任职要求
- 计算机科学、工程或相关专业的学士或硕士学位。
- 至少六年机器学习平台和机器学习应用开发经验。
- 具备 Python、C++ 或同等语言的高级编程能力。
- 具有使用 Lance、PyTorch、Ray Data 或同类技术的经验。
- 证明具备在快速发展的环境中构建可扩展、可靠基础设施的成功经验，并能与建模团队中的机器学习工程师合作。
- 深入理解系统设计中的权衡，并具备在跨职能团队之间建立共识的沟通能力。
- 具备模型训练、模型优化或大规模数据处理流水线方面的经验。
- 具备出色的分析推理和问题解决能力。
- 具备清晰的书面和口头沟通能力，包括向非技术利益相关者解释复杂技术概念的能力。
- 为满足美国国家安全和出口管制要求，可能需要核实居住地、美国人身份和/或公民身份。
福利待遇
- 提供平等就业机会，重视包容、创业精神和创新，不因性别、种族、年龄、性取向、宗教、残障或身份而区别对待。