Senior AI Engineer, NeMo Retriever - Model Optimization and MLOps jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 3 weeks ago

Senior AI Engineer, NeMo Retriever - Model Optimization and MLOps

NVIDIA is at the forefront of the AI revolution, focusing on technological advancements in intelligent assistants and information retrieval. The Senior AI Engineer will work on developing and optimizing machine learning models and MLOps practices, contributing to the creation of multimodal AI applications.

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

Responsibilities

Develop and maintain NIMs that containerize optimized models using OpenAPI standards using Python or an equivalent performant language
Work closely with partner teams to understand requirements, build & evaluate POCs, and develop roadmaps for production-level tools
Enable development of integrated systems - AI Blueprints that provide a unified, turnkey experience
Help build and maintain our Continuous Delivery pipeline with the goal of moving changes to production faster and safer while ensuring key operational standards
Provide peer reviews to other specialists, including feedback on performance, scalability, and correctness

Qualification

Python programmingDeep Learning frameworksMLOps technologiesCloud infrastructureNLPGenerative AIContinuous learningTeam collaborationProblem-solving

Required

Bachelor's or Master's Degree program in Computer Science, Computer Engineering, or a related field (or equivalent experience)
8+ years of demonstrated experience in a similar or related role
Python programming expertise with Deep Learning (DL) frameworks such as PyTorch
Experience delivering software in a cloud context and is familiar with the patterns and processes of handling cloud infrastructure
Knowledge of MLOps technologies such as Docker-Compose, Containers, Kubernetes, Helm, data center deployments, etc
Familiarity with ML libraries, especially PyTorch, TensorRT, or TensorRT-LLM
Excellent in-depth hands-on understanding of NLP, LLM, MLLM, Generative AI, and RAG workflows
Self-starter with a passion for growth, enthusiasm for continuous learning, and sharing findings across the team
Extremely motivated, highly passionate, and curious about new technologies

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