Machine Learning Engineer, GeForce G-Assist

NVIDIA Corporation

Santa Clara (CA)

On-site

USD 152,000 - 288,000

Full time

9 days ago

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

NVIDIA Corporation in Santa Clara is seeking an experienced systems engineer to help scale the GeForce G-Assist on-device AI platform. You will work across SLM/VLM models, optimize local inference, and implement RAG systems to ground responses in context.

Your work will cover C/C++ performance-critical code paths, Python tooling, and collaborating with product and platform teams to balance model capability, performance, and user experience.

Qualifications

  • 5+ years of validated experience in system software or related field; MS or higher (or equivalent).
  • Strong ability to read and write C/C++ in systems-level environments; proficiency in Python.
  • Hands-on experience with llama.cpp or similar local inference frameworks.
  • Experience evaluating Small Language Models, including task-based and conversational testing.
  • Knowledge of SLM and VLM architectures, retrieval technologies, and agentic AI patterns.

Responsibilities

  • Evaluate and improve Small Language Models used in GeForce G-Assist with emphasis on accuracy and conversational reliability.
  • Identify and mitigate conversation and context contamination (state drift, prompt leakage, retrieval cross-talk).
  • Work with SLM and VLM architectures to support text and multimodal interactions.
  • Collaborate on hybrid architectures that combine local SLMs with cloud-based models.
  • Read, write, and optimize C/C++ code in performance-critical paths.
  • Design and integrate retrieval-augmented generation (RAG) systems grounded in context.

Skills

C/C++
Python
SLM architectures
VLM architectures
Retrieval technologies
Agentic AI patterns

Education

Master's degree or higher in CS/DS/Engineering

Tools

llama.cpp

Job description

At NVIDIA, we’re building GeForce G-Assist — an on-device AI assistant that combines Small Language Models (SLMs), retrieval systems, and hybrid cloud capabilities to deliver responsive, context-aware assistance inside the GeForce ecosystem. We work closely across engineering and product teams to ensure G-Assist performs reliably in real-world scenarios.

What you'll be doing:

Together, we focus on how models behave in production, not just on benchmarks.

Evaluate and improve Small Language Models used in GeForce G-Assist, with an emphasis on accuracy, robustness, and conversational reliability.

Identify and mitigate conversation and context contamination, including state drift, prompt leakage, and retrieval cross-talk.

Work with SLM and VLM architectures to support text and multimodal interactions.

Collaborate on hybrid architectures that combine local SLMs with cloud-based models.

We value engineers who enjoy thinking across the full system—from model behavior to runtime performance.

Optimize local inference using llama.cpp, including quantization, memory usage, and performance tuning.

Read, write, and optimize C/C++ code in performance-critical paths.

Design and integrate retrieval-augmented generation (RAG) systems that ground responses in system and user context.

Support agentic AI workflows, enabling planning, tool use, and multi-step execution.

What we need to see:

5+ years of validated experience in system software or a related field, with an M.S. or higher degree in Computer Science, Data Science, Engineering, or a related field (or equivalent experience).

We’re looking for teammates who enjoy solving real problems, learning as they go, and collaborating in a tight-knit environment.

Strong ability to read and write C/C++ code in systems-level or performance-sensitive environments, along with proficiency in Python.

Hands-on experience with llama.cpp or similar local inference frameworks.

Hands-on experience evaluating Small Language Models, including task-based and conversational testing, with an understanding of conversation dynamics, long-context behavior, and contamination challenges.

Knowledge of SLM and VLM architectures and their trade-offs, experience with retrieval technologies and language-model integration, and familiarity with agentic AI patterns such as tool use and planning.

Ways to stand out from the crowd:

Experience contributing to language or multimodal models that power user-facing products, features, or workflows.

A track record of collaborating with product, platform, or systems teams to balance model capability, performance, and user experience.

Demonstrated ability to translate user needs or feedback into measurable improvements in model behavior or system reliability.

We are widely considered to be one of the technology world's most desirable employers, and as a result, we have some of the most forward-thinking and hardworking people in the world working for us.

If you're passionate, creative, and driven, we'd love to have you join the team. With competitive salaries and a generous benefits package, we are considered to be one of the technology world's most desirable employers.

We have some of the most forward-thinking and hardworking people in the world working for us, and due to unprecedented growth, our exclusive engineering teams are rapidly growing.

We want to hear from you if you're a creative and autonomous engineer with a real passion for technology.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4. You will also be eligible for equity and benefits. Applications for this job will be accepted at least until October 20, 2026. This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

NVIDIA pioneered accelerated computing. Today, our AI infrastructure powers global intelligence, transforming every industry.

Learn more about NVIDIA.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Machine Learning Engineer
Machine Learning Engineer

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 152,000 - 242,000
Equity
Benefits package
Senior Systems Software Engineer - Machine Learning
Senior Systems Software Engineer - Machine Learning

NVIDIA • Santa Clara (CA)

On-site
USD 184,000 - 288,000
Senior Software Engineer - AI Inference Performance
Senior Software Engineer - AI Inference Performance

NVIDIA Corporation • Santa Clara (CA)

Hybrid
USD 184,000 - 357,000
Senior Deep Learning Software Engineer, Inference
Senior Deep Learning Software Engineer, Inference

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 184,000 - 357,000
Equity
Benefits
Senior System Software Engineer, Agentic Retrieval
Senior System Software Engineer, Agentic Retrieval

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 184,000 - 357,000
Senior Deep Learning Software Engineer, Inference
Senior Deep Learning Software Engineer, Inference

NVIDIA • California

Hybrid
USD 152,000 - 288,000
Equity
Benefits
Hybrid work model
Senior Developer Technology Engineer - Edge Agentic AI
Senior Developer Technology Engineer - Edge Agentic AI

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 152,000 - 288,000
Equity
Benefits
Competitive salary
Senior AI Engineer, AI tools
Senior AI Engineer, AI tools

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 224,000 - 357,000
Equity
Benefits
Senior Staff Engineer - Employee Productivity
Senior Staff Engineer - Employee Productivity

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 200,000 - 322,000
Equity
Health benefits
Accelerated Computing GPU Product Manager
Accelerated Computing GPU Product Manager

NVIDIA Corporation • Santa Clara (CA)

On-site
USD 168,000 - 328,000