Staff Machine Learning Engineer
Core
Design and build large-scale deep learning infrastructure and platforms for distributed model training to handle massive throughput and low-latency predictions across the open web, social networks, video, and mobile apps.
Role type
Staff Machine Learning Engineer (Technical Lead & Architect)
Builds
Core AI/ML-based services, large-scale deep learning infrastructure, and distributed model training platforms
Domain
Advertising technology, data science, distributed systems
Deliverable
production ML models | infrastructure
Required skills
System design for complex ML systems, deep learning architecture, production-grade Python/Go/Java/C++, large-scale data pipeline design, CI/CD for AI/ML services, mentorship of senior engineers
Preferred skills
Experience with ranking infrastructures, ambiguity resolution, cross-functional technical leadership
Technologies
PyTorch, JAX, TensorFlow, CUDA, NCCL, RDMA, Databricks, Spark, Kubernetes, Kafka, Prometheus
Responsibilities
Drive architectural vision and system design for core AI/ML services; Act as technical lead for complex multi-quarter initiatives; Design and build large-scale deep learning infrastructure; Partner with Product Management and executives to align ML capabilities with business strategy; Mentor and guide senior and mid-level data scientists and engineers; Establish standard methodologies and lead architecture design reviews
Seniority
Staff, technical pillar with strategic ownership and mentorship
