Staff / Principal Machine Learning Engineer

9 Hours ago • 6 Years + • $240,000 PA - $385,000 PA

Job Summary

Job Description

Inworld is seeking Staff and Principal level Machine Learning Engineers to research, design, and implement Machine Learning systems for generative AI products and agentic frameworks. Responsibilities include researching cutting-edge ML models, developing production-scale infrastructure, and providing mentorship. The role involves working with LLMs, diffusion-based synthesis models, and other modern AI stacks, aiming to optimize real-time interactive experiences. The engineer will work on projects such as implementing new attention algorithms for LLM serving, designing a high-level API for creating AI voices, and optimizing transformer-based speech recognition models. The ideal candidate will be results-oriented, possess strong problem-solving skills, and be familiar with modern AI tools and technologies.
Must have:
  • 6 years of software development experience in Python or C++.
  • 4 years of experience applying ML algorithms in NLP/speech/action-planning.
  • Fluency with ML frameworks like PyTorch, TensorFlow, or JAX.
Good to have:
  • Experience in fine-tuning and evaluating LLMs such as LLaMA, Mistral, etc.
  • Knowledge of working with embedded systems and/or running ML on edge devices.

Job Details

view open roles

Why Join Inworld

Inworld is the leading provider of AI technology for real-time interactive experiences, with a $500 million valuation and backing from top tier investors including Intel Capital, Microsoft’s M12 fund, Lightspeed Venture Partners, Section 32, BITKRAFT Ventures, Kleiner Perkins, Founders Fund, and First Spark Ventures.

Inworld provides the market’s best framework for building production ready interactive experiences, coupled with dedicated services to optimize specific stages of development – from design and development, to ML pipeline optimization and custom compute infrastructure. We help developers bring their AI engines in-house with a framework optimized for real-time data ingestion, low latency, and massive scale. Inworld powers experiences built by Ubisoft, NVIDIA, Niantic, NetEase Games and LG, among others, and has partnerships with key industry players such as Microsoft Xbox, Epic Games, and Unity. 

Inworld was recognized by CB Insights as one of the 100 most promising AI companies in the world in 2024 and was named among LinkedIn's Top Startups of 2024 in the USA.

We are seeking Staff and Principal level Machine Learning Engineers with extensive experience in researching, designing, and implementing Machine Learning systems. You will be at the forefront of building generative AI products and agentic frameworks that utilize a wide range of modern AI stacks, including LLMs, diffusion-based synthesis models, and many more.

Qualifications

  • 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.

You May Be a Good Fit If You Have

  • 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.

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.

Representative Projects

  • Exploring the impact on quality and latency of using a new attention implementation algorithm within the LLM serving infrastructure.
  • Designing and implementing a high-level public API to enable developers to create new AI voices and use them at runtime.
  • Exploring various on-device matrix multiplication backends to optimize transformer-based speech recognition models for a specific device.

In-office location: Mountain View, California, United States. Candidates must be based in the SF Bay Area or willing to relocate (you will be working on-site in our South Bay office a few days a week). 

The US base salary range for this full-time position is $240,000 - $385,000. In addition to base pay, total compensation includes equity and benefits. Within the range, individual pay is determined by work location, level, and additional factors, including competencies, experience, and business needs. The base pay range is subject to change and may be modified in the future.

Inworld Jobs Privacy

Similar Jobs

ION - Data Engineer, Italy

ION

Italy (Hybrid)
6 Months ago
Moloco - Machine Learning Engineer

Moloco

Redwood City, California, United States (On-Site)
8 Hours ago
ByteDance - AI Research Engineer, Large Language Model (Applied Machine Learning)

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
GoMotive - Computer Vision Engineer

GoMotive

(Remote)
1 Day ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Unity - Staff Software Engineer

Unity

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
10 Hours ago
Trendyol - Data Science Professionals - Trendyol GO

Trendyol

İzmir, İzmir, Türkiye (Hybrid)
5 Months ago
NVIDIA - Senior Software Engineer - Automated Parallel Programming

NVIDIA

Santa Clara, California, United States (Remote)
3 Months ago
Canva - Senior Machine Learning Engineer - Specialist Platform and Experience

Canva

Surry Hills, New South Wales, Australia (Remote)
1 Month ago
Google - Software Engineer III, Generative AI, Google Workspace

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
Scale AI - Machine Learning Research Scientist / Engineer

Scale AI

San Francisco, California, United States (On-Site)
1 Day ago
ByteDance - Research Scientist in ML Systems

ByteDance

Seattle, Washington, United States (On-Site)
6 Months ago
Aristocrat Gaming - Automation and AI Engineer (Unity & Backend Technologies)

Aristocrat Gaming

Barcelona, Catalonia, Spain (Hybrid)
2 Weeks ago
Rackspace Technology - AI/ML Architect

Rackspace Technology

Vietnam (Remote)
2 Months ago
ByteDance - Research Scientist, Reinforcement Learning

ByteDance

San Jose, California, United States (On-Site)
6 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Mountain View, California, United States

Trek - Bicycle Service Technician

Trek

Madison, Wisconsin, United States (On-Site)
2 Months ago
Rackspace Technology - Support Data Engineer II

Rackspace Technology

San Antonio, Texas, United States (Remote)
1 Month ago
Google - Software Engineer III, Google Cloud Platforms

Google

Kirkland, Washington, United States (On-Site)
5 Months ago
singularity6 - Senior Data Scientist

singularity6

Los Angeles, California, United States (Hybrid)
19 Hours ago
Meta - Postdoctoral Researcher, Embodied AI (PhD)

Meta

Seattle, Washington, United States (On-Site)
5 Months ago
Postman - People Ops Communication Contractor

Postman

San Francisco, California, United States (Hybrid)
1 Day ago
Interactive Brokers - Automated Quality Assurance Engineer

Interactive Brokers

Greenwich, Connecticut, United States (Hybrid)
10 Hours ago
Tap Nation - Unity Developer

Tap Nation

New York, New York, United States (Remote)
5 Months ago
Snap Mobile INC - Account Executive

Snap Mobile INC

Baton Rouge, Louisiana, United States (On-Site)
1 Day ago
Inworld AI - Staff / Principal Machine Learning Engineer - USA

Inworld AI

Mountain View, California, United States (Remote)
5 Months ago

Get notifed when new similar jobs are uploaded

Similar Category Jobs

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

About The Company

Inworld powers AI-driven gameplay for video games and immersive experiences. 


Whether you’re looking to unlock novel gameplay, create content at scale, improve player immersion, or future proof your AI infrastructure, Inworld helps uplevel your game development with AI.


Inworld has worked with Xbox, Ubisoft, NVIDIA, NetEase Games, Niantic, LG UPlus, Alpine Electronics, and indie game developers to create AI-driven experiences. We’re backed by top-tier investors including Section 32, Intel Capital, Microsoft’s M12 fund, BITKRAFT Ventures, Kleiner Perkins, Founders Fund,and First Spark Ventures. 


The Inworld product suite includes: 


Inworld Engine powers real-time experiences with groundbreaking game mechanics, dynamic NPCs, and worlds that evolve with each action. AI NPCs can learn and adapt, deliver nuanced performances, perceive the world around them, and autonomously initiate actions based on players' decisions. 


Inworld Studio consists of a suite of tools that enhance game design. Using AI to streamline workflows, the Studio enables developers to workshop, draft, and outline storylines, narratives, dialogue, quests, and more.  


Inworld Core is our custom solution for future-proof AI infrastructure, including custom models, training, serving and security.

Mountain View, California, United States (On-Site)

Mountain View, California, United States (On-Site)

Vancouver, British Columbia, Canada (On-Site)

Vancouver, British Columbia, Canada (On-Site)

Mountain View, California, United States (On-Site)

Mountain View, California, United States (On-Site)

Mountain View, California, United States (On-Site)

Mountain View, California, United States (Hybrid)

Vancouver, British Columbia, Canada (On-Site)

Mountain View, California, United States (Hybrid)

View All Jobs

Get notified when new jobs are added by Inworld AI

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug