Senior Machine Learning Engineer

Unity Technologies SF; Mountain View (CA); On-site; USD 210,000 - 273,000; Full time

Benefits offered by this job

Health insurance

Stock options

Pension plan

Vacation days

Parental leave

Commute subsidy

Wellbeing programs

Training & development

Volunteering

Job summary

Unity Technologies SF is seeking a Senior Machine Learning Engineer to architect and scale agentic AI within the Unity engine. This onsite Mountain View role emphasizes designing goal-directed systems, memory and world modeling, and real-time performance.

You will lead major initiatives, mentor teams, and drive the roadmap for agentic AI across games and simulations while ensuring safety and quality in production.

Qualifications

• MS or Ph.D. in CS, ML, robotics, or equivalent practical experience.
• 5+ years of hands-on experience developing ML systems in production, ideally in real-time or interactive environments.
• A strong background in reinforcement learning, AI planning, decision-making under uncertainty, multi-agent systems.

Responsibilities

• Design, implement, and scale AI agent systems in Unity environments.
• Build reusable agentic frameworks such as planning modules or memory systems.
• Develop AI models for Unity's agentic system to support tasks in Unity games.
• Integrate ML models within real-time engine constraints, optimizing latency and determinism.
• Lead initiatives like hierarchical reinforcement learning for scalable behaviors.
• Collaborate with teams to expand use cases across gaming and simulation.
• Contribute to Unity's agentic AI roadmap with modularity and performance.
• Establish best practices for model quality and AI safety in real-time systems.
• Mentor engineers and researchers to strengthen innovation and collaboration.

Skills

reinforcement learning

AI planning

multi-agent systems

decision-making under uncertainty

fine-tuning

goal-directed behavior

environment-aware decision-making

Education

MS or Ph.D. in Computer Science, ML, Robotics

Job description

Unity Technologies SF is hiring a Senior Machine Learning Engineer to help build foundational agentic AI systems inside the Unity engine. This onsite role in Mountain View, CA focuses on designing, scaling, and integrating AI agent systems and models for real-time games and simulations, including planning, memory and world modeling, and ML/AI safety.

You will work on systems that support goal-directed behavior, task planning and execution, and environment-aware decision-making. The role also emphasizes technical leadership: you will drive high-impact initiatives, help define long-term direction for Unity's agentic AI offerings, and mentor engineers and researchers while strengthening a culture of innovation, excellence, and collaboration.

What you'll do

• Design, implement, and scale AI agent systems in Unity environments to enable goal-directed behavior, task planning and execution, and environment-aware decision-making.
• Build reusable agentic frameworks such as planning modules, memory systems, or policy adaptation layers that creators can extend in their own projects.
• Develop AI models for Unity's AI agentic system to support tasks involved in building Unity games.
• Integrate ML models within real-time game engine constraints, optimizing for latency, compute, and determinism.
• Lead major initiatives, including hierarchical reinforcement learning for scalable behaviors, AI planning and goal inference for NPCs and simulations, and agent memory and world modeling for persistent, believable behavior.
• Collaborate with internal and external teams to expand use cases across gaming, simulation, and interactive storytelling.
• Contribute to Unity's agentic AI roadmap, ensuring modularity, usability, and performance across platforms.
• Establish and champion best practices for model quality, simulation-based evaluation, and ML/AI safety within real-time systems.
• Mentor engineers and researchers, supporting a technical culture of innovation, excellence, and collaboration.

What you bring

• MS or Ph.D. in Computer Science, Machine Learning, Robotics (or equivalent practical experience).
• 5+ years of hands-on experience developing ML systems in production, ideally in real-time or interactive environments.
• A strong background in one or more of the following: reinforcement learning, fine-tuning, decision-making under uncertainty, AI planning, or multi-agent systems.

Compensation and location

Location: Mountain View, CA (onsite)

Salary: USD 210,300 to 273,400 per year

Benefits

• 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

English communication requirement

This position requires the incumbent to have sufficient knowledge of English, including frequent and regular communication with colleagues and partners worldwide whose common language is English.

Privacy

For application details, review the Prospect and Applicant Privacy Policies. Contact: [e-mail]