Staff Machine Learning Developer

Summary

Join Unity's Ads Applied Research team as a Staff Machine Learning Developer and build the next-generation ML platform for Unity Ads. You will design, develop, and deploy scalable ML systems to optimize ad delivery, bidding, and user engagement. This role involves contributing to the technical roadmap, translating research into practical solutions, and collaborating with cross-functional teams. You will focus on building and enhancing platform capabilities for training, versioning, deploying, and serving high-scale ad-tech models using technologies like Golang and NVIDIA Triton Inference Server. The ideal candidate possesses extensive experience in designing, deploying, and maintaining ML systems at scale, expertise in Python and ML frameworks, and strong communication skills. This position requires a background in Computer Science, Machine Learning, or a related field.

Requirements

• 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

Responsibilities

• 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

Preferred Qualifications

• 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

Benefits

$186,700 — $280,100 CAD