Staff / Principal Machine Learning Engineer - USA

$240,000 - $385,000/Yr

Inworld AI - Mountain View, CA

Full-time - Senior

Mountain View, CA

Publishing Industries

About the position

Inworld is the leading AI engine for games and interactive media, recognized for its innovative approach and significant backing from top-tier investors. We are seeking Staff and Principal level Machine Learning Engineers who will be instrumental in building generative AI products and agentic frameworks. This role involves utilizing a wide range of modern AI stacks, including LLMs and diffusion-based synthesis models, to create interactive and personalized gaming experiences. You will be at the forefront of advancing Inworld's AI capabilities, working on projects that impact the gaming industry significantly.

Responsibilities

• Research and experiment with cutting-edge ML models and techniques to advance Inworld's AI capabilities.
• Develop production-scale infrastructure to train, evaluate, and serve ML models efficiently.
• Provide guidance and mentorship to junior engineers, fostering a culture of learning and collaboration.

Requirements

• BA/BS degree or higher in Computer Science, Engineering, or a similar technical field.
• 6 years of experience in software development in one or more programming languages such as Python or C++.
• 4 years of experience in applying ML algorithms in natural language processing and/or speech processing and/or action-planning domains.
• Fluency with ML frameworks such as PyTorch, TensorFlow, or JAX.
• Experience in fine-tuning and evaluating LLMs, such as LLaMA, Mistral, Qwen, etc., is a significant plus.
• Knowledge of working with embedded systems and/or running ML on edge devices is a big plus.

Nice-to-haves

• Significant software engineering experience and a results-oriented mindset.
• Strong problem-solving and analytical skills, with a proactive approach to tackling challenges.
• Familiarity with modern AI tools and technologies, such as transformers, diffusion models, reinforcement learning.
• Ability to work collaboratively in a fast-paced environment with shifting priorities.
• Passion for learning and staying up-to-date with the latest advancements in machine learning research and its applications.

Benefits

• Equity in the company
• Comprehensive health insurance
• Flexible working hours
• Remote work options
• Professional development opportunities