Machine Learning Engineer, Performance Tooling
Core
Design and build reliable, self-service performance tools that scale across models, hardware targets, and development workflows to measure, predict, and optimize AI performance.
Role type
Senior IC machine learning performance engineer
Builds
Self-service performance tooling and monitoring systems for model builds and training runs
Domain
AI/ML infrastructure and performance optimization
Deliverable
production ML models
Required skills
performance engineering, profiling, roofline analysis, latency optimization, throughput optimization, workload root-cause analysis, Python, PyTorch, deep learning model development, data analysis
Preferred skills
end-to-end tool ownership, quantitative communication
Technologies
Python, PyTorch
Responsibilities
Design and build performance tools; define standards for measuring and predicting AI performance; model theoretical platform peaks and identify efficiency losses; predict latency, memory, utilization, and compute cost; own monitoring and regression alerting; collaborate with training and runtime engineers to establish performance targets.
Seniority
Senior, hands-on IC
