Senior Machine Learning Engineer, AI Authoring

Unity Technologies; Austin (TX); Remote; USD 192,000 - 289,000; Full time

Job summary

A leading technology company is seeking a Senior Machine Learning Engineer (AI Authoring) to lead the development of foundational AI systems in real-time 3D applications. This role involves designing AI agents, collaborating across teams, and driving advancements in AI integration in gaming. Candidates should have an advanced degree and significant ML experience in production. The salary range is $192,600—$288,800 USD.

Qualifications

• 5+ years of hands-on experience developing ML systems in production.
• Strong system design skills and ability to architect ML systems.
• Proficiency in at least one general-purpose programming language.

Responsibilities

• Design, implement, and scale AI agents for game code generation.
• Collaborate to unlock innovative use cases in gaming and simulations.
• Champion best practices for model quality and ML safety.

Skills

Machine Learning

AI Development

Python

C++

Collaborative Skills

Education

Advanced degree in Computer Science or related field

Tools

PyTorch

TensorFlow

Job description

Remote, USA

Senior Machine Learning Engineer, AI Authoring

Location

Remote, USA

Department

Engineering

Role Description

At Unity, we’re shaping the future of real-time 3D development by enabling creators to harness intelligent, autonomous systems. Our next-generation agentic AI - goal-directed, adaptive, and capable of reasoning and planning in dynamic environments - is transforming how games and simulations come to life.

We’re looking for a Staff Machine Learning Engineer to lead the development of these foundational AI systems within the Unity engine, empowering creators to build smarter, more responsive in-game agents and simulations. Your work will directly drive innovation, allowing developers and designers to integrate sophisticated AI behavior seamlessly into their worlds.

What You'll Be Doing

• Design, implement, and scale AI agents that assist in generating game code and executing goal-directed behaviors.
• Define orchestration logic for single- and multi-agent systems
• Develop reusable agentic frameworks - planning modules, memory systems, or policy adaptation layers - for creator extensibility.
• Lead high-impact initiatives, including hierarchical reinforcement learning for scalable behaviors, AI planning and goal inference frameworks for NPCs and simulations, and agent memory and world modeling systems for persistent, believable behavior.
• Collaborate across internal and external teams to unlock innovative use cases in gaming, simulation, and interactive storytelling.
• Contribute to the long-term vision and roadmap for Unity’s agentic AI offerings, 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.

What We're Looking For

• Advanced degree (MS or Ph.D.) in Computer Science, Machine Learning, or a related field.
• 5+ years of hands-on experience developing ML systems in production, ideally within 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; ability to work effectively with cross-functional teams including engine developers, designers, and product managers.

You Might Also Have

• Familiarity with game development workflows.
• Publications or open-source contributions in reinforcement learning, agent-based modeling, or related areas.
• Experience with behavior trees, utility AI, or hierarchical task planning.
• Passion for the gaming industry and AI applications in games.

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.

The salary range for this role is $192,600—$288,800 USD.