Principal AI and ML Engineer — AI for Networking

Sorry, this job was removed at 08:10 p.m. (CST) on Wednesday, Jul 16, 2025
Be an Early Applicant
Santa Clara, CA
In-Office
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
The Role

NVIDIA redefines what’s possible. NVIDIA has been reinventing computer graphics, PC gaming, and accelerated computing for 30 years. It is a unique legacy of innovation that’s fueled by great technology and amazing people. Today, we’re tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, generative AI, robots, and self-driving cars that can understand the world. Doing what’s never been done before takes vision, innovation, and the world’s best talent. As an NVIDIAN, you’ll be immersed in a diverse, supportive environment where everyone is inspired to do their best work.

Our company is at the forefront of technological innovation, and we are dedicated to driving efficiency and optimizing the performance of our infrastructure both on-prem and cloud. Join us in this exciting endeavor! We are seeking a highly skilled Principal AI/ML Engineer to join our dynamic team to build the next generation of IT Networking space and help lead the team through a major technology transformation into running AI on-prem and build infrastructure by integrating Enterprise ready platforms while building a solid foundation with automation. We are looking for a passionate engineer who will solve networking problems with AI.

What You’ll Be Doing:

  • Architect and implement infrastructure platforms tailored for AI/ML workloads, with a focus on scaling private cloud environments to support high-throughput training, inference, and Agentic workflows and pipelines.

  • Lead initiatives in Generative AI systems design, including Retrieval-Augmented Generation (RAG), LLM fine-tuning, semantic search, and multi-modal data processing.

  • Build and optimize ML systems for document understanding, vector-based retrieval, and knowledge graph integration using advanced NLP and information retrieval techniques.

  • Design and develop scalable services and tools to support GPU-accelerated AI pipelines, leveraging Kubernetes, Python/Go, and observability frameworks.

  • Mentor and collaborate with a multidisciplinary team of network engineers, automation engineers, AI and ML scientists, product managers, and multiple domain experts.

  • Build and drive adoption of emerging AIOPs technologies, integrating AI Agents, RAGs, and LLMs using MCP workflows to streamline automation, performance tuning, and large-scale data insights.

What We Need to See:

  • 10+ years of engineering experience with at least 5 years leading initiatives in ML infrastructure, AI systems, or applied NLP/LLM development.

  • 5+ years of experience in Networking and infrastructure.

  • Bachelor’s, Master’s, or Ph.D. in Computer Science, Engineering, Machine Learning, or a related field (or equivalent experience).

  • Deep expertise with:

    • Generative AI concepts such as embeddings, RAG, semantic search, and transformer-based LLMs

    • MCP workflows and Agentic ecosystem

    • Vector databases (e.g., FAISS, Pinecone, Weaviate) and data pipelines

    • Programming in Python (preferred) and/or Go, and software engineering best practices

  • Experience deploying and tuning LLMs using techniques like LoRA, QLoRA, and instruction tuning.

  • Strong understanding of infrastructure automation pipeline (Terraform, Ansible, Salt), monitoring (Prometheus, Grafana), and DevOps tools.

  • Hands-on experience working with petabyte-scale datasets, schema design, and distributed processing.

  • Strong background in working with infrastructure related data collections and logs related to network data. Ability to run simulations of network state with AI tools.

Ways to Stand Out From the Crowd:

  • Experience building multi-hop RAG systems with self-consistency and chain-of-thought prompting.

  • Prior leadership in designing AI platforms used for large-scale enterprise search, document intelligence, or recommendation systems.

  • Contributions to open-source ML/AI tools or active participation in the AI research community.

  • Familiarity with knowledge graph construction and reasoning systems as well as demonstrated ability to communicate complex ML concepts to executive and cross-functional stakeholders.

  • Strong knowledge of automation pipeline and infrastructure configuration and observability tools like: BigPanda, Splunk, Storm, Netbox/Nautobot and different open-source automation tooling

  • Strong knowledge of different "network operating systems" like Arista EOS, Cumulus, Cisco NX-OS, Sonic, SRLinux as well as excellence in Infrastructure or Network as a Code automation frameworks

NVIDIA is widely 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. If you're creative and autonomous, we want to hear from you!

#LI-Hybrid

The base salary range is 248,000 USD - 391,000 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse 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.

Similar Jobs

Bevi Logo Bevi

Account Executive

Greentech • Hardware • Healthtech • Internet of Things
Remote or Hybrid
California, USA
227 Employees
130K-161K Annually

Motorola Solutions Logo Motorola Solutions

Inside Sales Representative

Artificial Intelligence • Hardware • Information Technology • Security • Software • Cybersecurity • Big Data Analytics
Remote or Hybrid
California, USA
23000 Employees
24-29 Hourly
Remote or Hybrid
9 Locations
213000 Employees
27-41 Hourly

Block Logo Block

Behavioral Insights Lead - Bitcoin

Blockchain • eCommerce • Fintech • Payments • Software • Financial Services • Cryptocurrency
In-Office or Remote
8 Locations
12000 Employees
89K-168K Annually
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Clara, CA
21,960 Employees
Year Founded: 1993

What We Do

NVIDIA’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, NVIDIA is increasingly known as “the AI computing company.”

Similar Companies Hiring

LayerOne Thumbnail
Software • Information Technology • Artificial Intelligence
New York, NY
15 Employees
Scotch Thumbnail
Software • Retail • Payments • Fintech • eCommerce • Artificial Intelligence • Analytics
US
25 Employees
Idler Thumbnail
Artificial Intelligence
San Francisco, California
6 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account