Staff Machine Learning Engineer

Integral Ad Science is seeking a Staff Machine Learning Engineer to join the Data Science Team in New York, NY. In this onsite role, you will provide technical leadership and operational ownership for large-scale machine learning systems and infrastructure supporting core AI/ML services.

Role Responsibilities

• Lead the architectural vision and system design for core AI/ML-based services.
• Serve as technical lead for complex, multi-quarter initiatives from inception through global deployment.
• Design and build large-scale deep learning infrastructure and distributed training platforms 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 senior and mid-level data scientists and engineers.
• Define standard methodologies, set code quality expectations, and lead architecture design reviews.
• Work with large-scale AI training infrastructure components including accelerators, network fabrics, CUDA, NCCL, and RDMA, along with big data ecosystems such as Databricks, Spark, Kubernetes, Kafka, and Prometheus.
• Design, develop, and support robust CI/CD pipelines for AI/ML services to ensure smooth transitions from research to production endpoints.

Required Qualifications

• 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 demonstrated record of operating at a Lead, Principal, or Staff level.
• Deep expertise designing complex ML systems, including ranking infrastructures and data pipelines capable of handling massive throughput.
• Production-hardened experience with frameworks including PyTorch, JAX, or TensorFlow.
• Strong production-grade programming skills in Python, Go, Java, or C++, with a strong understanding of software design principles and algorithms.
• Proven ability to operate in ambiguity and independently drive large, unstructured problems to technical resolution.
• A track record of collaborating with, elevating, and mentoring engineering talent.

Technologies

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

Compensation and Location

Location: New York, NY (onsite)

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

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

IAS Pay Transparency

• Primary Location: US - New York, NY
• Primary Location Base Pay Range: $135,100.00 - $231,600.00 Annual
• Additional Locations: USA - Remote
• Additional Locations Pay Range: 119,000.00 - 204,000.00 USD Annual

Agency / Third-Party Recruiters

• IAS does not accept unsolicited resumes or candidate profiles.
• If you are interested in becoming an IAS recruiting partner, send an email introducing your company to [email protected].