Senior Staff Machine Learning Engineer — Agentic AI jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 6 days ago

Senior Staff Machine Learning Engineer — Agentic AI

NVIDIA is known as 'the AI computing company,' and they are seeking a Senior Staff Machine Learning Engineer to join their Enterprise AI team. The role involves developing and productionizing advanced AI systems, focusing on intelligent solutions that enhance enterprise operations and productivity.

AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Develop Intelligent AI Solutions – Leverage NVIDIA AI technologies and GPUs to build pioneering NLP and Generative AI solutions—such as Retrieval-Augmented Generation (RAG) pipelines and agentic workflows—that solve real-world enterprise and supply-chain problems
Own Key AI Features – Drive the end-to-end development of LLM-powered applications, chatbots, and optimization engines that improve organizational efficiency and resilience
Design Robust ML Architectures – Create machine-learning and combinatorial-optimization designs targeting high-impact challenges across employee productivity, engineering efficiency, AIOps, and supply-chain operations, etc
Collaborate Across NVIDIA – Work closely with product, research, and engineering teams to translate requirements into ML solutions and deliver measurable business outcomes
Mentor & Share Best Practices – Guide junior engineers and peers on ML design patterns, code quality, and experiment methodology

Qualification

Machine LearningNatural Language ProcessingDeep LearningPythonPyTorchTensorFlowTransformersCollaborationMentoringProblem Solving

Required

Master's or Ph.D. in Computer Science, Operations Research, Industrial Engineering, or a related field, or equivalent experience
10+ years designing, building, and deploying machine-learning models and systems in production with 12+ years industry experience
Solid understanding of transformers, attention mechanisms, and modern NLP / LLM techniques; experience fine-tuning or prompting large language models
Strong Python plus deep-learning frameworks such as PyTorch or TensorFlow; familiarity with CUDA-accelerated libraries (e.g., TensorRT-LLM) is a plus
Proven track record to take a significant ML component or feature from concept to production and collaborate effectively with multi-functional teams

Preferred

Agentic AI Mastery – Practical experience with frameworks such as LangChain or LangGraph and a deep understanding of multi-step reasoning and planning
LLM Inference Optimization – Expertise in accelerating LLM inference (e.g., KV caching, quantization) to achieve low latency at scale
End-to-End ML Systems Ownership – A portfolio showing full lifecycle ownership, from data ingestion to monitoring and continuous improvement
Research Impact – Publications or patents that advance NLP or enterprise AI

Benefits

Equity
Benefits

Company

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

H1B Sponsorship

NVIDIA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)

Funding

Current Stage
Public Company
Total Funding
$4.09B
Key Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity

Leadership Team

leader-logo
Jensen Huang
Founder and CEO
linkedin
leader-logo
Michael Kagan
Chief Technology Officer
linkedin
Company data provided by crunchbase