Staff Machine Learning Infrastructure Engineer, Embedding Platform

For our client, we are seeking a Staff Machine Learning Infrastructure Engineer Embedding Platform to join the team of a leader in the Media space. This role will lead work at the intersection of data, AI-enabled capabilities, and scalable technology delivery. You will work across engineering, product, operations, and business stakeholders to translate complex requirements into practical technology solutions. The position offers the opportunity to influence architecture, execution quality, and the technology capabilities that enable long-term growth within a Media environment.Location: Remote - US based candidates only, no visa sponsorship availableCompensation: $253,300 – $354,600 annuallyResponsibilities Architect and lead the development of next-generation, large-scale machine learning techniques Define and execute the ML strategy to enhance personalization and recommendation quality Lead research on scalable ML systems and real-time model adaptation Partner with ML infrastructure teams to build distributed training systems Establish and optimize real-time serving architectures for low-latency inference Collaborate with teams in Feed Ranking, Ads, Content Understanding, and Core ML on model integration Mentor and guide engineering staff, promoting a culture of excellence and innovation Drive technical discussions and contribute to long-term ML planningQualifications 8+ years of experience in machine learning engineering, focused on large-scale ML systems and personalization Expertise in modern deep learning architectures, such as sequence models and foundational models Deep understanding of complex multi-entity relationships within large-scale ML applications Proven ability to design and optimize scalable ML architectures for distributed training and real-time inference Strong software engineering skills in Python, C++, or similar languages, with experience in cloud-based ML pipelines Demonstrated leadership in mentoring and influencing cross-functional teams Experience with A/B testing and model evaluation frameworks in production systems Excellent communication skills for presenting complex ML conceptsBenefits Comprehensive Healthcare Benefits and Income Replacement Programs 401k with Employer Match Global Benefit programs supporting lifestyle, professional development, and caregiving Family Planning Support Gender-Affirming Care Mental Health & Coaching Benefits Flexible Vacation & Paid Volunteer Time Off Generous Paid Parental LeaveOur client is an equal opportunity employer. We encourage you to apply even if you don’t meet every qualification—your background could be exactly what this team needs.