Staff / Principal Machine Learning Engineer

1 Week ago • 6 Years + • $240,000 PA - $385,000 PA

Job Summary

Job Description

This role involves researching, designing, and implementing Machine Learning systems, specifically for building generative AI products and agentic frameworks. The responsibilities include researching and experimenting with ML models, developing production-scale infrastructure, and mentoring junior engineers. The job description emphasizes the need for strong software engineering skills, problem-solving abilities, and a collaborative approach to working in a fast-paced environment. Representative projects include exploring attention implementation algorithms within LLM serving infrastructure and optimizing transformer-based speech recognition models.
Must have:
  • 6 years of software development experience in Python or C++.
  • 4 years of experience applying ML algorithms in NLP, speech processing, or action-planning.
  • Fluency with ML frameworks such as PyTorch or TensorFlow.
  • BA/BS degree in Computer Science or similar field.
Good to have:
  • Experience fine-tuning and evaluating LLMs.
  • 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

Ubisoft - Lead R&D Scientist

Ubisoft

Shanghai, Shanghai, China (On-Site)
1 Month ago
Reality Games - Machine Learning Engineer - Monopoly World

Reality Games

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
2 Months ago
Synechron - Java Developer with Github CoPilot / AI Code Generator

Synechron

Dallas, Texas, United States (On-Site)
1 Week ago
ByteDance - Research Scientist, Reinforcement Learning

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
ByteDance - Research Scientist Graduate (Foundation Model Speech & Audio Generation)

ByteDance

Seattle, Washington, United States (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Kwalee - Machine Learning Engineer

Kwalee

Royal Leamington Spa, England, United Kingdom (On-Site)
2 Months ago
ByteDance - Imaging System Architect

ByteDance

San Jose, California, United States (On-Site)
3 Weeks ago
Upwork - Senior Lead Machine Learning Engineer

Upwork

(Remote)
2 Weeks ago
Demandbase - Data Science / Machine Learning Engineer Intern

Demandbase

(Remote)
1 Week ago
Google - Customer Engineer, Applied and Generative AI, Google Cloud

Google

Singapore, Singapore (On-Site)
3 Weeks ago
WebFX - Entry Level Software Engineer

WebFX

Harrisburg, Pennsylvania, United States (On-Site)
6 Months ago
ByteDance - Research Scientist in Large Model System

ByteDance

Seattle, Washington, United States (On-Site)
6 Months ago
Google - Technical Program Manager III, AI/ML, Cloud AI Systems

Google

Austin, Texas, United States (On-Site)
4 Weeks ago
Nextbrain - Computer Vision Engineer

Nextbrain

Bengaluru, Karnataka, India (On-Site)
5 Months ago
N-iX - Middle Data Science/AI Engineer

N-iX

Langenfeld, North Rhine-Westphalia, Germany (Hybrid)
1 Week ago

Get notifed when new similar jobs are uploaded

Jobs in Mountain View, California, United States

Thales - Pre-Sales Solution Expert (Digital Payment and Card Issuance Services)

Thales

Austin, Texas, United States (Hybrid)
1 Week ago
ByteDance - AMS - Accounts Payable and Reimbursements Shared Services Center Leader

ByteDance

Los Angeles, California, United States (Hybrid)
2 Months ago
Mattel  Inc  - American Girl Experience Associate

Mattel Inc

Dallas, Texas, United States (On-Site)
2 Months ago
Critical mass - Creative Engineer

Critical mass

San Jose, California, United States (On-Site)
1 Week ago
lifechruh - StaffKids Teacher: Seasonal Support

lifechruh

Edmond, Oklahoma, United States (On-Site)
1 Year ago
Sleeper - Backend Engineer

Sleeper

San Francisco, California, United States (Remote)
1 Month ago
Fortra - Business Development Representative - Strategic Accounts

Fortra

United States (On-Site)
2 Weeks ago
NVIDIA - Manager, Developer Technology, Data Compression

NVIDIA

California, United States (Hybrid)
3 Months ago
Scout - Director, Interior and Seats

Scout

Novi, Michigan, United States (On-Site)
2 Weeks ago
GoDaddy - Professional Services Guide

GoDaddy

Tempe, Arizona, United States (Hybrid)
1 Week 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