Staff Machine Learning Developer

The opportunity

We are seeking a Staff Machine Learning Developer to make significant contributions to the next-generation of ML systems for Unity Ads. In this role, you will focus on building the next-generation ML platform that powers our high-scale ad-tech models, from training to inference. You will be deeply involved in the design, development, and deployment of ML models and the underlying platform, collaborating closely with product, engineering, and research teams to advance our ML capabilities.

What you'll be doing

• Design, develop, and deploy scalable, production-grade ML systems and the platform components that support them, to optimize ad delivery, bidding strategies, and user engagement.
• Contribute significantly to the technical roadmap and design of our ML platform, focusing on solutions for training and inference of high-scale ad-tech models.
• Translate state-of-the-art research into practical, scalable solutions for our ML platform and models, ensuring Unity remains at the forefront of ML performance and innovation in the ad tech space.
• Partner cross-functionally with data science, product, and engineering teams to align ML platform capabilities and model development with strategic business goals.
• Collaborate on defining and refining requirements for data infrastructure, MLOps tooling, and processes to support advanced ML initiatives and the ML platform.
• Promote and implement best practices for model development, testing, deployment, monitoring, and overall MLOps within the new platform.
• Specifically focus on building and enhancing the ML platform capabilities for training, versioning, deploying, and serving high-scale ad-tech models, leveraging technologies such as Golang for performance-critical services and NVIDIA Triton Inference Server for optimized model serving.

What we're looking for

• Background in Computer Science, Machine Learning, Statistics or a related field.
• Extensive hands-on experience designing, deploying, and maintaining ML systems at scale, including managing cloud infrastructure on GCP and orchestrating workloads with Kubernetes.
• Strong track of record building ML platforms and infrastructure, including MLOps best practices — from data pipelines and model training workflows to serving and observability — with hands-on experience using tools like NVIDIA Triton Inference Server.
• Expertise in Python and familiarity with production-grade ML frameworks (such as PyTorch or TensorFlow), as well as modern ML architectures (transformer-based models, embeddings, retrieval-augmented pipelines).
• Familiarity with robust MLOps practices, data pipelines, and cloud ML deployment, including model serving and monitoring.
• Excellent communication and collaboration skills, with an ability to articulate technical concepts to both technical and non-technical audiences.

You might also have

• Experience with Golang for building high-performance ML systems/infrastructure.
• A forward-thinking mindset, always seeking opportunities to innovate and improve processes.
• A passion for mentoring and supporting teammates, fostering a culture of collaboration and shared success.
• Experience in ad tech or other performance-critical domains like recommender systems or real-time personalization.

Additional information

• International relocation support is not available for this position.

Life at Unity

As of the third quarter of 2024, more than 70% of the top 1,000 mobile games were made with Unity, and in 2024, Made with Unity applications had an average of 3.7 billion downloads per month. For more information, please visit Unity.

Unity is a proud equal opportunity employer. We are committed to fostering an inclusive, innovative environment and celebrate our employees across age, race, color, ancestry, national origin, religion, disability, sex, gender identity or expression, sexual orientation, or any other protected status in accordance with applicable law. Our differences are strengths that enable us to support the growing and evolving needs of our customers, partners, and collaborators. If you have a disability that means there are preparations or accommodations we can make to help ensure you have a comfortable and positive interview experience, please fill out this form to let us know.

This position requires the incumbent to have a sufficient knowledge of English to have professional verbal and written exchanges in this language since the performance of the duties related to this position requires frequent and regular communication with colleagues and partners located worldwide and whose common language is English.

Headhunters and recruitment agencies may not submit resumes/CVs through this website or directly to managers. Unity does not accept unsolicited headhunter and agency resumes. Unity will not pay fees to any third-party agency or company that does not have a signed agreement with Unity.

Your privacy is important to us. Please take a moment to review our Prospect Privacy Policy and Applicant Privacy Policy. Should you have any concerns about your privacy, please contact us at DPO@unity.com.

#SEN #LI-AR1

*Note: Certain locations require a good faith disclosure of the base salary range for the role. The actual salary for the successful candidate may differ based on location, experience, and other job-related factors.