Senior Machine Learning Engineer, AI Automation

Unity New York, NY $148,700 - $223,100
Full Time Senior Level 8+ years

Posted 2 weeks ago

Interested in this position?

Upload your resume and we'll match you with this and other relevant opportunities.

Upload Your Resume

About This Role

This role involves modernizing Unity's core infrastructure and building backend systems for next-generation AI agents that revolutionize campaign experiences within a world-class ad-tech ecosystem. The engineer will solve complex system challenges at massive scale and directly influence Unity’s revenue growth.

Responsibilities

  • Design and deploy intelligent agentic systems that integrate large language models (LLMs) with enterprise data, tools, and workflows using frameworks like LangChain, LlamaIndex, and Semantic Kernel
  • Develop Retrieval-Augmented Generation (RAG) applications using tools like Azure AI Search, vector databases, and secure enterprise connectors to deliver contextual insights
  • Build and iterate on conversational agents that solve real-world problems, meet stakeholder needs, and deliver measurable business value
  • Deliver high-impact features by collaborating across teams, leading through ambiguity, and aligning technical solutions with business goals
  • Drive quality and performance through automated testing, monitoring, and data-driven evaluation of success criteria, user adoption, and operational efficiency
  • Mentor teammates and contribute to a culture of innovation, technical excellence, and continuous learning

Requirements

  • Bachelor’s/Master’s level degree in computer science or relevant engineering-related field or equivalent experience
  • 8-10+ years of software engineering experience
  • 1-2+ years working on AI-powered systems or products
  • Foundational knowledge of or hands-on experience with LLM orchestration frameworks (e.g., LangChain, LlamaIndex, Semantic Kernel) and prompt engineering using LLM APIs
  • Expertise in LLM integration, prompt engineering, and the technical trade-offs of different model architectures for specific ad-tech use cases
  • Hands-on experience with vector stores (e.g., Pinecone, Milvus, Weaviate), embedding models, and data orchestration for context-aware AI
  • Familiarity with the Model Context Protocol (MCP)
  • Strong proficiency in Java, Go, or Python
  • Background in building large-scale distributed systems and real-time data pipelines (Kafka, Flink, or Spark)

Qualifications

  • Bachelor’s/Master’s level degree in computer science or relevant engineering-related field or equivalent experience
  • 8-10+ years of software engineering experience, including 1-2+ years working on AI-powered systems or products

Nice to Have

  • Understanding of auction mechanics, bidding algorithms, or advertiser delivery funnels

Skills

Python * Java * Go * LangChain * Kafka * LlamaIndex * Spark * Pinecone * Flink * Weaviate * Semantic Kernel * Azure AI Search * Milvus *

* Required skills

Benefits

Well-being programs
Work-life balance support

About Unity

Unity [NYSE: U] is the leading platform to create and grow games and interactive experiences. Creators, ranging from game developers to artists, architects, automotive designers, infrastructure experts, filmmakers, and more, use Unity to bring their imaginations to life across multiple platforms, fr...

Technology
View all jobs at Unity →