Senior Machine Learning Infrastructure Engineer

Unity; Mountain View (CA); On-site; USD 184,000 - 249,000; Full time

Benefits offered by this job

Health insurance

Employee stock ownership

Commuting subsidy

Generous vacation and personal days

Parental leave

Job summary

Unity is seeking a Senior Machine Learning Infrastructure Engineer for its Vector Ads team in Mountain View. You will design, build, and operate real-time ML model serving infrastructure to support Unity’s global ads platform, delivering low-latency, high-throughput services.

You’ll collaborate with ML engineers to productionize models, optimize deployments, and improve observability and cost efficiency in a remote-first environment.

Qualifications

• Experience building and operating ML infrastructure or model serving systems in production.
• Proficiency in Golang or Python with strong systems engineering fundamentals.
• Hands-on experience with Kubernetes and container orchestration at scale.
• Familiarity with ML serving frameworks such as Ray Serve, Triton, TorchServe, or similar.

Responsibilities

• Design, build, and maintain the infrastructure that serves ML models in real-time across Unity’s ads ecosystem.
• Build and operate scalable model serving pipelines — owning latency, throughput, and reliability in a high‑QPS production environment.
• Partner with ML engineers to productionize models, manage model deployments, and improve iteration speed.
• Improve observability, performance, and cost‑efficiency of ML serving infrastructure.
• Contribute to architectural decisions around feature serving, model versioning, and inference optimization.

Skills

ML infra experience

Python and Go

Distributed systems

Model serving

System reliability

Remote collaboration

Tools

Kubernetes

Ray Serve

Triton

TorchServe

Terraform

Prometheus

Grafana

OpenTelemetry

Job description

The opportunity

Unity is looking for a Senior Machine Learning Infrastructure Engineer to join our Vector Ads team, where we build the real-time systems that power Unity's global advertising platform. This is a high-scale, low-latency environment — processing billions of requests daily to deliver fast, relevant ads to players around the world.

You’ll build and operate the infrastructure that brings ML models from training into production, ensuring our ranking, bidding, and targeting systems run reliably at scale. This is a great opportunity for an engineer who’s excited to work at the intersection of ML systems and distributed infrastructure, collaborate across teams, and have direct impact on how machine learning shapes the player and advertiser experience.

What you’ll be doing

• Design, build, and maintain the infrastructure that serves ML models in real-time across Unity’s ads ecosystem
• Build and operate scalable model serving pipelines — owning latency, throughput, and reliability in a high‑QPS production environment
• Partner with ML engineers to productionize models, manage model deployments, and improve iteration speed
• Improve observability, performance, and cost‑efficiency of ML serving infrastructure
• Contribute to architectural decisions around feature serving, model versioning, and inference optimization

What we’re looking for

• Experience building and operating ML infrastructure or model serving systems in production
• Proficiency in Golang or Python, with strong systems engineering fundamentals
• Hands‑on experience with Kubernetes and container orchestration at scale
• Familiarity with ML serving frameworks such as Ray Serve, Triton, TorchServe, or similar
• Understanding of distributed systems, API design, and system reliability
• Strong collaboration and communication skills in a remote‑first environment

You might also have

• Experience with feature stores, feature pipelines, or online/offline feature serving
• Background in ad tech, real‑time bidding, or programmatic advertising systems
• Familiarity with infrastructure‑as‑code such as Terraform
• Experience with observability tooling — Prometheus, Grafana, OpenTelemetry
• Background with real‑time data pipelines, caching layers, or low‑latency serving systems

Additional information

• Relocation support is not available for this position

Benefits

At Unity, we want our team members to thrive. We offer a wide range of benefits designed to support well‑being and work‑life balance.

Please note: Benefits eligibility, specific offerings, and coverage vary based on the country and employment status.

While specific benefits vary, here are some of the ways we strive to take care of our eligible team members globally:

• Comprehensive health, life, and disability insurance
• Commute subsidy
• Employee stock ownership
• Competitive retirement/pension plans
• Generous vacation and personal days
• Support for new parents through leave and family‑care programs
• Office food snacks
• Mental Health and Wellbeing programs and support
• Employee Resource Groups
• Global Employee Assistance Program
• Training and development programs
• Volunteering and donation matching program

Life at Unity

Unity [NYSE: U] is the world’s leading game engine, powering play for more than 3 billion consumers each month. The top mobile games in the world, the most played PC indie titles, the most innovative console games, and virtually all of the top XR and Web Games are developed, deployed, and grown in Unity. Unity also enables teams across industries like automotive, manufacturing, and healthcare to design, simulate, and collaborate in 3D — closing the gap between ideas and reality. For more information, please visit www.unity.com.

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.

*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.

Gross pay salary$183,700—$248,600 USD