Skip to main content
Back to jobs

Senior Machine Learning Infrastructure Engineer

External
Unity logoUnity · Mountain View, CA
Full-timeOn-site2w ago
API DesignCachingGrafanaKubernetesMachine LearningObservability
Cover LetterConnect

Prepare for this interview

Elite

AI-generated questions, company research, and talking points tailored to this role


About the role

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.

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

Requirements

  • 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.Life at UnityHeadhunters 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.YourHealth insurancePaid time offRemote work optionsEquity / stock options

Your Match

How well this role fits your profile.

Company Intel

What employees say

Worked at Unity? Share your experience

Interested in this role?

Apply on the company's website.

Cover LetterConnect