Unity Logo

Unity

Senior Machine Learning Infrastructure Engineer

Reposted 8 Days Ago
Be an Early Applicant
Hybrid
San Francisco, CA, USA
184K-249K Annually
Senior level
Hybrid
San Francisco, CA, USA
184K-249K Annually
Senior level
The role involves designing and maintaining ML infrastructure, building serving pipelines, collaborating with ML engineers, and enhancing system performance for Unity's advertising platform.
The summary above was generated by AI

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.

Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

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 Web site 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 [email protected].

#SEN #LI-ED1

*Note: This range reflects the anticipated base salary for this position. Beyond base salary, this role may be eligible for equity awards and participation in our company incentive plans (such as annual discretionary bonuses or sales commissions). The final offer amount will depend on several factors, including geographic location and the candidate’s relevant experience, professional background, and skill set.
 
Gross pay salary
$183,700$248,600 USD
HQ

Unity San Francisco, California, USA Office

San Francisco, CA, United States

Unity San Francisco, California, USA Office

Work at Unity includes a ton of flexibility, depending on your role, the needs of your team and ultimately the way you and your team work best together.

Similar Jobs

4 Days Ago
In-Office
San Francisco, CA, USA
170K-190K Annually
Senior level
170K-190K Annually
Senior level
Artificial Intelligence • Logistics • Robotics • Software
Design, build, and operate large-scale ML infrastructure and GPU compute clusters for computer vision and multi-modal models. Own end-to-end pipelines from data ingestion and training to low-latency cloud deployment, orchestration, monitoring, and model performance evaluation. Collaborate with research and product teams to productionize models across warehouse environments.
Top Skills: AirflowC++CudaDeepspeedFlyteGpuGrpcKafkaNvidia Triton Inference ServerPythonPyTorchQuantizationRos2TemporalTensorFlowTensorflow ServingTensorrtTorchserve
7 Days Ago
In-Office or Remote
150K-250K Annually
Senior level
150K-250K Annually
Senior level
Transportation
Design, build, and operate scalable MLOps infrastructure for data ingestion, distributed training, deployment, CI/CD, experiment tracking, and monitoring across cloud R&D and production. Collaborate with ML, autonomy, robotics, and software teams to deliver architecture, PoCs, and production-ready ML pipelines that support real-time, safety-critical systems.
Top Skills: AirflowAWSAzureCi/CdGCPGitGpuHorovodKubeflowMetaflowMlflowPythonPytorch DdpRaySagemaker
25 Days Ago
In-Office
San Francisco, CA, USA
176K-220K Annually
Senior level
176K-220K Annually
Senior level
Edtech • Enterprise Web • HR Tech • Software
Build and operate shared ML/AI infrastructure: data pipelines, feature stores, training, model serving and LLM platform work. Scale inference and training (GPU, batching, autoscaling), enable evaluation and post-training workflows, and partner with AI, data science, and product teams to productionize models and improve platform reliability and developer experience.
Top Skills: AirflowApache BeamAutoscalingAWSBatchingBigQueryCi/CdDataflowDockerEmbeddingsFeature StoresGCPGoGpu InferenceKubernetesLlmsModel ServingPythonSparkStreaming PipelinesTerraformTypescript

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account