Staff Machine Learning Engineer

Integral Ad Science, Inc. is hiring a Staff Machine Learning Engineer for the Data Science Team to lead and oversee large-scale AI/ML systems and architectures that support business predictions across the open web, social networks, video, and mobile applications. This onsite role in San Francisco includes technical leadership and end-to-end ownership from CI/CD through production deployment.

Responsibilities

• Set the architectural vision and own system design for core AI/ML-based services.
• Serve as technical lead for multi-quarter initiatives, guiding work from inception through global deployment.
• Design and build large-scale deep learning infrastructure and platforms for distributed model training, focused on low-latency and high availability at enterprise scale.
• Collaborate with Product Management, Core Engineering, and executive stakeholders to align ML capabilities with business strategy.
• Convert abstract product requirements into concrete technical designs.
• Mentor and guide senior and mid-level data scientists and engineers.
• Define standard methodologies, establish code quality expectations, and lead architecture design reviews.
• Work with large-scale AI training infrastructure components, including accelerators, network fabrics, CUDA, NCCL, and RDMA, as well as big data ecosystems such as Databricks, Spark, Kubernetes, Kafka, and Prometheus.
• Design, develop, and support robust CI/CD pipelines for AI/ML services to move models reliably from research into production endpoints.

Requirements

• PhD or Master’s degree in a technical field (for example, computer science, mathematics, or statistics) or equivalent years of experience.
• 7+ years of machine learning experience in industry with a demonstrated record at Lead, Principal, or Staff level.
• Deep expertise in designing complex machine learning systems, ranking infrastructures, and data pipelines capable of massive throughput.
• Production-hardened experience with frameworks including PyT orch, JAX, or T ensorFlow.
• Strong production-grade programming skills in Python, Go, Java, or C++, with a deep understanding of software design principles and algorithms.
• Proven ability to operate effectively in ambiguous environments and drive large unstructured problems to technical resolution.
• A history of collaborating with, elevating, and mentoring engineering talent.

Technologies

• PyT orch, JAX, T ensorFlow
• Python, Go, Java, C++
• CUDA, NCCL, RDMA
• Databricks, Spark, Kubernetes, Kafka, Prometheus

Benefits

• Paid time off
• Health insurance (medical, dental, vision)
• PPO, HSA and FSA options
• 401k with employer matching contributions
• Eligible for an annual bonus and/or other incentive plans

Pay and Location

Location: San Francisco, CA (onsite)

Salary: USD 135,100 - 231,600 per year

IAS Pay Transparency:

• Primary Location: US - New York, NY
• Primary Location Base Pay Range: $135,100.00 - $231,600.00 Annual
• Additional Locations: US - Chicago, IL; US - San Francisco, CA
• Additional Locations Pay Range: 119,000.00 - 204,000.00 USD Annual