Staff Machine Learning Engineer

New Today

Overview Remote, USA Staff Machine Learning Engineer Role description The opportunity 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. This role is perfect for someone passionate about pushing the boundaries of AI, multi-agent systems, reinforcement learning, and intelligent decision-making in complex, real-time environments. 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. Additional information Relocation support is not available for this position. Life at Unity We offer a wide range of benefits designed to support employees' well-being and work-life balance. You can read more about them on our career page (https://unity.com/careers). 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, from mobile, PC, and console, to spatial computing. As of the third quarter of 2024, more than 70% of the top 1,000 mobile games were made with Unity, and in 2024, Made with Unity applications had an average of 3.7 billion downloads per month. For more information, please visit Unity (https://www.unity.com). This position requires the incumbent to have a sufficient knowledge of English to have professional verbal and written exchanges in this language since the performance of the duties related to this position requires frequent and regular communication with colleagues and partners located worldwide and whose common language is English. Headhunters and recruitment agencies may not submit resumes/CVs through this website or directly to managers. Unity does not accept unsolicited headhunter and agency resumes. Unity will not pay fees to any third-party agency or company that does not have a signed agreement with Unity. Your privacy is important to us. Please take a moment to review our Prospect Privacy Policy and Applicant Privacy Policy. If you have any concerns about your privacy, please contact us. Gross pay salary $192,600—$288,800 USD
#J-18808-Ljbffr
Location:
Boston, MA, United States
Salary:
$250,000 +
Job Type:
FullTime
Category:
Engineering

We found some similar jobs based on your search