Staff Machine Learning Engineer
Core
Build infrastructure, tooling, and optimization systems to enable efficient training, evaluation, deployment, and operation of machine learning models at scale.
Role type
Staff Machine Learning Engineer (ML Infrastructure)
Builds
Scalable ML training and inference platforms, developer tooling, and resource management systems
Domain
Internet / Machine Learning Infrastructure
Deliverable
production ML models
Required skills
Python, systems languages (Go/C++/Rust/Java), distributed systems, ML infrastructure, performance engineering, debugging, profiling
Preferred skills
large-scale recommendation/ranking/generative AI, distributed training frameworks (PyTorch Distributed/Ray/Spark), GPU architecture, cloud cost optimization
Technologies
PyTorch Distributed, Ray, Tensorflow, Spark
Responsibilities
Design systems to improve ML training and inference efficiency; Develop tooling for debugging, profiling, and monitoring; Optimize GPU and resource utilization; Partner with teams to identify bottlenecks; Build benchmarking frameworks and dashboards; Lead cross-functional initiatives for platform productivity and strategy
Seniority
Staff, hands-on IC with strategic scope
