Staff Machine Learning Engineer

Integral Ad Science, Inc. is hiring a Staff Machine Learning Engineer to help lead the Data Science Team’s core AI/ML services. This onsite role in New York, NY focuses on building and scaling large-scale machine learning systems and infrastructure, from distributed deep learning platforms to production CI/CD pipelines. You’ll have the opportunity to shape technical direction, mentor engineers, and translate business goals into reliable technical implementations.

Compensation: $135,100 - $231,600 per year (Primary Location: US - New York, NY). Additional locations include USA - Remote with a pay range of $119,000 - $204,000 per year.

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 large-scale deep learning infrastructure for distributed model training, prioritizing low latency and high availability at enterprise scale.
• Partner with Product Management, Core Engineering, and executive stakeholders to align ML capabilities with business strategy.
• Convert abstract product needs into concrete technical designs.
• Mentor and guide senior and mid-level data scientists and engineers.
• Establish standard methodologies, define code quality expectations, and lead architecture design reviews.
• Work with large-scale AI training infrastructure components including accelerators, network fabrics, CUDA, NCCL, and RDMA, plus big data ecosystems such as Databricks, Spark, Kubernetes, Kafka, and Prometheus.
• Design, develop, and support robust CI/CD pipelines for AI/ML services, ensuring models move from research to production endpoints.

What you bring

• A PhD/Master’s degree in a technical field such as computer science, mathematics, statistics, or equivalent years of experience.
• 7+ years of machine learning experience in industry, with a track record operating 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 such as PyTorch, JAX, or TensorFlow.
• Strong production-grade programming skills in Python, Go, Java, or C++, plus a solid understanding of software design principles and algorithms.
• Ability to thrive in ambiguity, take on large unstructured problems, and independently drive them to technical resolutions.
• A strong history of collaborating with, elevating, and mentoring engineering talent.

Tools you’ll work with

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

Compensation and benefits

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

Additional information

Attention Agency/3rd Party Recruiters: IAS does not accept any unsolicited resumes or candidate profiles. If you are interested in becoming an IAS recruiting partner, please send an email introducing your company to [email protected].