
Unity Technologies
Senior Machine Learning Engineer, ML Infrastructure - Offline
Senior Machine Learning Engineer, ML Infrastructure - Offline
发布于 大约 2 个月前普通员工/个人贡献者
上海市
高级经验
全职员工
仅现场办公
学历未注明
软件工程
分布式计算
CUDA
Flyte
GPU
ML基础设施
PyTorch
Torch Compilation
AI 估算 · 35k–55k
资深ML工程师岗位,技术栈前沿(Ray、PyTorch、GPU优化),上海市场薪资竞争力强,但Unity作为成熟公司,薪资偏稳健。
职位详情
关于这个职位
该职位是Unity Vector团队的资深机器学习工程师,专注于离线ML基础设施的设计与优化
你将负责构建大规模训练数据管道、分布式训练工作流,并提升ML管道的可靠性和可扩展性
适合有深厚分布式系统背景和ML基础设施经验的工程师
最低要求
具有使用分布式计算框架(如Ray、Spark、Flink)的经验,熟悉Ray生态系统(Ray Data、Ray Train)用于分布式数据处理和模型训练
具有构建和优化大规模分布式ML训练管道的经验,包括Torch Compilation、Quantization、CUDA、GPU kernel优化等
具有构建训练数据生成、数据集准备或ML特征管道基础设施的经验
具有设计和运维生产级数据管道的深厚经验
精通Python编程,具有大规模分布式工作负载经验
熟悉现代数据基础设施(数据湖、数据仓库、编排系统、流处理平台)
具备强大的系统思维,能够推理分布式系统中的性能、可扩展性、可靠性和成本权衡
具有领导技术方向、跨团队影响架构决策的能力(非正式权力)
工作职责
设计并运维大规模数据管道,生成用于机器学习训练和实验的训练数据集
开发支持分布式训练工作流的基础设施,使用PyTorch、Ray Data、Ray Train等技术
将ML管道与工作流编排系统(如Flyte、Airflow或类似系统)集成,实现可靠的多阶段训练工作流
通过数据集验证、监控和自动化测试提高ML管道的可复现性和可观测性
优化分布式计算系统在数据处理和模型训练中的性能和资源利用率
与ML工程师紧密合作,支持高效的大规模实验和模型迭代
领导架构改进,确保离线ML管道保持可扩展、可靠和成本高效
AI 洞察
优缺点分析
优点
- 技术栈前沿:使用Ray、PyTorch、CUDA等最新技术,技能积累价值高
- 公司平台强:Unity作为上市巨头,资源丰富,业务稳定
- 影响范围广:所建基础设施支撑全公司的ML工作,能见度高
- 技术深度要求高:需深入理解GPU优化、分布式训练等底层知识
- 适合已有分布式系统经验、喜欢构建基础设施、对ML性能优化有热情的工程师
缺点 / 挑战
- 工作强度可能较大:需要运维大规模分布式系统,on-call压力
角色解读
- 技术专家路线:深耕ML基础设施,成为分布式训练或数据管道领域的权威
- 架构师路线:负责整个离线平台的技术架构,影响公司级ML策略
- 管理路线:有机会带领团队,从IC过渡到技术管理
- 设计并构建大规模离线ML平台,包括训练数据生成和分布式训练管道
- 使用Ray、PyTorch等框架优化分布式训练工作流,提升性能和资源利用率
- 与ML工程师协作,支持大规模实验和模型迭代,确保管道可靠可观测
- 熟练掌握Python及分布式计算框架(Ray、Spark、Flink),能调优GPU训练性能
- 具备数据基础设施经验(数据湖、编排系统),能设计生产级数据管道
- 有系统思维,能在性能、可扩展性、成本之间做权衡
申请策略
- 了解Unity的业务场景(游戏引擎、AI驱动决策),在面试中体现对业务的理解
- 准备好系统设计案例,展示如何设计可扩展的离线ML管道
- 重点突出使用Ray、Spark等框架构建大规模分布式系统的项目经验
- 展示GPU优化、PyTorch训练管道调优的具体成果(如吞吐量提升、成本降低)
- 强调数据管道设计中的系统思考和工程最佳实践
- 深入学习Ray生态(Ray Data、Ray Train)和PyTorch分布式训练
- 补充工作流编排工具(如Flyte、Airflow)的实际操作经验
面试指南
- 使用STAR方法(情境、任务、行动、结果)回答行为问题
- 对于设计问题,先明确需求(规模、延迟、可靠性),然后比较方案优劣,最后给出推荐
- 技术问题从原理出发,结合实际经验,注意权衡取舍
- 如何设计一个大规模分布式训练数据管道?
- 如何优化PyTorch分布式训练的性能?
- 描述一次你解决分布式系统可靠性问题的经历
- 如何使用Ray实现数据并行和模型并行?
- 在数据量增长时,如何保证管道可扩展性和成本可控?
职位点评
74
综合评分
前沿技术栈、强成长性,但WLB一般,适合技术驱动型人才。
从薪资福利、成长空间、工作节奏和岗位方向综合评估,方便横向比较。
更适合这类人
适合追求技术成长、愿意投入时间学习前沿技术、对分布式ML基础设施有热情的求职者。
表现最好
成长发展
相对薄弱
工作生活
薪资福利75
成长发展90
工作生活55
使命价值70
薪资福利
75中等
薪资估算处于市场高位,福利完善(如股票、保险、休假),但未明确具体数字,吸引力中等偏上。
薪资信号未披露(AI估算:35K-55K/月)
福利待遇Comprehensive health, life, and disability insurance、Commute subsidy、Employee stock ownership、Competitive retirement/pension plans、Generous vacation and personal days、Support for new parents through leave and family-care programs、Office food snacks、Mental Health and Wellbeing programs、Training and development programs
成长发展
90较高
技术栈前沿(Ray、PyTorch、GPU优化),有培训和发展项目,成长空间大。
技术前沿前沿/新兴技术
技术栈Ray、PyTorch、CUDA、Torch Compilation、GPU、Spark、Flink、Flyte
成长机会Training and development programs
业务类型profit_center
工作生活
55较低
仅现场办公,未明确WLB,可能涉及on-call,但提供通勤补贴。
工作模式仅现场办公
办公地点未明确
加班情况未提及(无法判断)
工作生活平衡Commute subsidy
使命价值
70中等
Unity游戏引擎影响力大,ML平台推动产品智能化,行业前景好。
行业发展稳定成熟行业
社会影响中性/一般
创新程度积极采用新技术
Unity Technologies 的其他在招职位
Senior Strategic Sales
Unity Technologies · 北京市AI 估算 · 30k-50kGPS - Programmatic Strategy Partnership Manager
Unity Technologies · 北京市AI 估算 · 35k-55kStaff Machine Learning Engineer, ML Infrastructure - Offline
Unity Technologies · 上海市AI 估算 · 40k-70kStaff Machine Learning Engineer, ML Infrastructure - Online
Unity Technologies · 上海市AI 估算 · 50k-80kSenior Machine Learning Engineer, ML Infrastructure - Online
Unity Technologies · 上海市AI 估算 · 30k-60k
相似职位推荐
Watch Jobs