Staff Machine Learning Engineer

Integral Ad Science, Inc. is seeking a Staff Machine Learning Engineer to join the Data Science team in New York, NY (remote). This role focuses on technical leadership for large-scale AI/ML services and production machine learning systems that support globally deployed models across open web, social, video, and mobile apps. If you enjoy shaping end-to-end architecture, mentoring engineering talent, and turning ambitious ideas into production outcomes, this is a strong fit.

What you’ll do

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

What you bring

• PhD/Master’s in a technical field such as computer science, mathematics, statistics, or equivalent years of experience.
• 7+ years of industry machine learning experience, with a track record operating at the Lead, Principal, or Staff level.
• Deep expertise designing complex ML systems, including ranking infrastructures and data pipelines capable of massive throughput.
• Production-hardened experience with frameworks including PyTorch, JAX, and/or TensorFlow.
• Strong production-grade programming skills in Python, Go, Java, or C++, plus solid software design principles and algorithms.
• Proven ability to operate in ambiguity, take large unstructured problems, and drive them to successful technical resolution.
• A strong history of collaborating with, elevating, and mentoring engineering talent.

Compensation and benefits

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

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

Tools you’ll work with

PyTorch, JAX, TensorFlow, Python, Go, Java, C++, CUDA, NCCL, RDMA, Databricks, Spark, Kubernetes, Kafka, Prometheus