Nvidia

Machine Learning Engineer, GeForce G-Assist

Nvidia
Apply
1 day ago
Santa Clara, CA, USASenior
H1B sponsor

Base Salary

$152k - $288k/yr

Responsibilities

  • Evaluate and improve Small Language Models for accuracy, robustness, conversational reliability, and production behavior.
  • Identify and mitigate state drift, prompt leakage, retrieval cross-talk, and other conversation or context contamination issues.
  • Work with SLM and VLM architectures to support text and multimodal interactions.
  • Collaborate on hybrid architectures combining local SLMs with cloud-based models.
  • 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 systems grounded in system and user context.
  • Support agentic AI workflows involving planning, tool use, and multi-step execution.
  • Collaborate with product, platform, and systems teams to balance model capability, performance, and user experience.

Requirements

  • At least 5 years of validated experience in system software or a related field.
  • M.S. or higher in Computer Science, Data Science, Engineering, or a related field, or equivalent experience.
  • Strong ability to read and write C/C++ in systems-level or performance-sensitive environments.
  • Proficiency in Python.
  • Hands-on experience with llama.cpp or similar local inference frameworks.
  • Hands-on experience evaluating Small Language Models through task-based and conversational testing.
  • 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.
  • Familiarity with agentic AI patterns such as tool use and planning.
  • Experience contributing to language or multimodal models for user-facing products, features, or workflows is preferred.
  • Experience collaborating with product, platform, or systems teams and translating user feedback into measurable model or reliability improvements is preferred.

Benefits

  • Eligible for equity and benefits.
  • Applications will be accepted at least until October 20, 2026.

Tech Stack

Categories

Nvidia

About Nvidia

10,000+ employees

Nvidia designs and sells GPUs and accelerated computing platforms for data centers, AI/ML, graphics, gaming, and automotive, monetizing through hardware, software platforms (CUDA, AI frameworks), and systems like DGX and networking. Customers include cloud providers, enterprises, researchers, and OEMs. Founded in 1993 and headquartered in Santa Clara, it is a public company traded on NASDAQ under NVDA.

Contact me