Staff Machine Learning Engineer, AI Authoring

Unity
San Francisco, CA
Job Description
Unity is seeking a Staff Machine Learning Engineer to develop foundational AI systems within the Unity engine, enabling creators to build intelligent agents. The role focuses on designing, implementing, and scaling AI agents for dynamic environments, driving innovation and integration of AI behavior into games and simulations. This is a hands-on position for someone passionate about AI, game development, and machine learning.

Requirements

  • Advanced degree (MS or Ph.D.) in Computer Science, Machine Learning, or a related field.
  • 7+ years of hands-on experience developing ML systems in production, ideally in real-time or interactive environments.
  • Experience with multi-agent systems, orchestration, or AI tool integration.
  • Strong background in one or more of the following: LLM, reinforcement learning, decision-making under uncertainty, AI planning, or multi-agent systems.
  • Proficiency in at least one general-purpose programming language (e.g., Python, C++, Go, Java) and familiarity with ML frameworks (e.g., PyTorch, TensorFlow).
  • Strong system design skills and the ability to architect performant, scalable ML systems.
  • Excellent communication and collaboration skills.
  • Familiarity with game development workflows.

Benefits

  • Employee well-being and work-life balance
  • Wide range of benefits
]]>