Machine Learning Engineer, Geforce G-assist

NVIDIA NVIDIA · Semiconductors · Santa Clara, CA

Machine Learning Engineer at NVIDIA focused on building GeForce G-Assist, an on-device AI assistant. The role involves evaluating and improving SLMs and VLMs, optimizing local inference (e.g., llama.cpp), designing RAG systems, and supporting agentic AI workflows. Requires strong C/C++ and Python skills, experience with local inference frameworks, and knowledge of SLM/VLM architectures and agentic AI patterns.

What you'd actually do

  1. 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.
  2. 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.
  3. Optimize local inference using llama.cpp, including quantization, memory usage, and performance tuning. Read, write, and optimize C/C++ code in performance-critical paths.
  4. 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.

Skills

Required

  • 8+ 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)
  • 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

Nice to have

  • 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

What the JD emphasized

  • on-device AI assistant
  • Small Language Models (SLMs)
  • SLM and VLM architectures
  • local inference
  • retrieval-augmented generation (RAG) systems
  • agentic AI workflows

Other signals

  • on-device AI assistant
  • Small Language Models (SLMs)
  • retrieval systems
  • hybrid cloud capabilities
  • local inference
  • agentic AI workflows