Senior Machine Learning Engineer, AI Authoring jobs in United States
info-icon
This job has closed.
company-logo

Unity ยท 1 month ago

Senior Machine Learning Engineer, AI Authoring

Unity Technologies is shaping the future of real-time 3D development by enabling creators to harness intelligent, autonomous systems. They are looking for a Senior Machine Learning Engineer to lead the development of foundational AI systems within the Unity engine, empowering creators to build smarter in-game agents and simulations.

3D TechnologyReal Time
check
H1B Sponsor Likelynote

Responsibilities

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

Qualification

Machine LearningReinforcement LearningMulti-Agent SystemsPythonAI PlanningSystem DesignCommunicationCollaboration Skills

Required

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

Preferred

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

Benefits

A wide range of benefits designed to support employees' well-being and work-life balance.

Company

Unity [NYSE: U] offers a suite of tools to create, market, and grow games and interactive experiences across all major platforms from mobile, PC, and console, to extended reality.

H1B Sponsorship

Unity has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (53)
2024 (55)
2023 (39)
2022 (81)
2021 (62)
2020 (78)

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
Jim Whitehurst
Chief Executive Officer
linkedin
leader-logo
John Riccitiello
CEO
linkedin
Company data provided by crunchbase