Staff / Principal AI Researcher - USA

Inworld; Mountain View (CA); Remote; USD 240,000 - 385,000; Full time

Job summary

Join a leading AI startup recognized for its innovative contributions to gaming and interactive media. As a Staff or Principal AI Researcher, you will be at the forefront of developing state-of-the-art machine learning models that enhance AI capabilities in gaming. This role offers the chance to collaborate with top-tier industry partners and contribute to groundbreaking projects that shape the future of AI. With a strong emphasis on research and development, you will have the opportunity to explore novel ideas, train large-scale models, and mentor junior team members. If you are passionate about pushing the boundaries of AI and enjoy working in a collaborative environment, this position is perfect for you.

Qualifications

• PhD or equivalent experience in Computer Science or related field.
• Strong background in machine learning research with publications.

Responsibilities

• Conduct cutting-edge research to advance AI models and capabilities.
• Develop and implement novel machine learning techniques.

Skills

Machine Learning Research

AI Model Development

Python Programming

Proficiency with ML Frameworks

Knowledge of AI Architectures

Empirical Research Skills

Education

PhD in Computer Science

Equivalent Experience in AI

Tools

PyTorch

TensorFlow

JAX

Job description

view open roles

Why Join Inworld

Inworld is the best-funded startup in AI and games 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 is the leading AI engine for games and interactive media. Inworld’s suite of AI components enables developers to build interactive, responsive, and personalized AI gaming experiences, orchestrate models to create intelligent game behaviors, and unlock enhanced productivity with AI-generated content. 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 also named among LinkedIn's Top Startups of 2024 in the USA.

We are seeking Staff and Principal level AI Researchers with extensive experience in researching, designing, and implementing machine learning models and algorithms. You will be at the forefront of advancing our AI capabilities by developing state-of-the-art models and contributing to cutting-edge research in artificial intelligence. Your work will involve exploring novel ideas, training large-scale models, and collaborating closely with other researchers and engineers to push the boundaries of what's possible in AI.

Qualifications

• PhD or equivalent experience in Computer Science, Machine Learning, Artificial Intelligence, or a related technical field.
• Strong background in machine learning research, demonstrated by publications or significant contributions to the field.
• Experience in training large models and working with state-of-the-art approaches in machine learning.
• Proficiency with ML frameworks such as PyTorch, TensorFlow, or JAX.
• Knowledge of advanced AI architectures, including transformers, diffusion models, and reinforcement learning techniques.
• Programming skills in Python or similar languages.
• Familiarity with pre-training, fine-tuning and evaluating LLMs such as LLaMA, Mistral, Qwen, etc., is a significant plus.

You May Be a Good Fit If You Have

• A track record of innovating and improving machine learning algorithms.
• A passion for understanding and developing model architectures, and enjoy performing careful empirical research.
• The ability to own and pursue a research agenda, choosing impactful problems and autonomously carrying out long-term projects.
• A desire to collaborate in a team environment, working together to make significant discoveries.
• An excitement about staying up-to-date with the latest advancements in machine learning research and its applications.
• A view that research and engineering are two sides of the same coin, engaging in both to achieve your goals.

Responsibilities

• Conduct cutting-edge research to advance our AI models and capabilities.
• Develop and implement novel machine learning techniques, experimenting with new architectures and algorithms.
• Train and evaluate large-scale ML models, optimizing for quality, latency and cost.
• Collaborate with cross-functional teams, sharing insights and integrating research findings into our products.
• Communicate results internally and publicly, contributing to the broader AI community.
• Mentor junior researchers and engineers, fostering a culture of learning and collaboration.

Representative Projects

• Exploring the impact of new attention mechanisms within transformer architectures to improve model quality and latency.
• Investigating methods for 3D animation synthesis by performing ablation study to thoroughly compare diffusion vs flow-matching based approaches.
• Developing algorithms for on-device AI, optimizing transformer-based models for specific hardware constraints.
• Researching and implementing techniques for creating new AI voices, enabling developers to generate and use them at runtime.

In-office location: Mountain View, CA, United States.

Remote location: United States.

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