Principal AI and ML Engineer - AI for Networking

NVIDIA • Santa Clara, CA • 6h ago

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 MPC 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

MPC 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 d emonstrated 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 e xcellence 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-Hyrbrid

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 (https://www.nvidia.com/en-us/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.

NVIDIA

Go to Apply

Share this job

Similar Jobs

Manufacturing Engineer

Zimmer Biomet • Campbell, CA • 41min ago

IT & Engineering

At Zimmer Biomet, we believe in pushing the boundaries of innovation and driving our mission forward. As a global medical technology leader for nearly 100 years, a patient's mobility is enhanced by a Zimmer Biomet produ...

41min ago

Go to Apply

Manufacturing Engineer

Zimmer Biomet • San Jose, CA • 48min ago

IT & Engineering

48min ago

Go to Apply

Staff Software Engineer, Parallel File System, AI/ML

Google • Sunnyvale, CA • 4h ago

IT & Engineering

Applicants in the County of Los Angeles: Qualified applications with arrest or conviction records will be considered for employment in accordance with the Los Angeles County Fair Chance Ordinance for Employers and the C...

4h ago

Go to Apply

Senior Software Engineer, GPU Performance, Google Scale

Google • Sunnyvale, CA • 4h ago

IT & Engineering

Minimum qualifications: Bachelor's degree or equivalent practice experience. 5 years of experience with software development in one or more programming languages, and with data structures/algorithms. 3 years of ex...

4h ago

Go to Apply

Software Engineer III, Front End, Google Cloud AI

Google • Sunnyvale, CA • 4h ago

IT & Engineering

Minimum qualifications: Bachelor's degree or equivalent practical experience. 2 years of experience with software development in one or more programming languages, or 1 year of experience with an advanced degree in...

4h ago

Go to Apply

Senior Staff Software Engineer, Platform Security

Google • Mountain View, CA • 4h ago

IT & Engineering

Minimum qualifications: Bachelor's degree or equivalent practical experience. 8 years of experience in software development, and with data structures/algorithms. 5 years of experience with design and architecture,...

4h ago

Go to Apply

Principal Software Engineer, Infrastructure (SASE)

Palo Alto Networks • Santa Clara, CA • 4h ago

IT & Engineering

Our Mission At Palo Alto Networks® everything starts and ends with our Mission: Being the cybersecurity partner of choice, protecting our digital way of life. Our vision is a world where each day is safer and more...

4h ago

Go to Apply

Senior Software Development Engineer-Maps Performance

Apple • Cupertino, CA • 4h ago

IT & Engineering

Senior Software Development Engineer-Maps Performance Cupertino, California, United States Software and Services Summary Posted: May 05, 2025 Weekly Hours: 40 Role Number: 200602794 Apple Maps team is seek...

4h ago

Go to Apply

Front End Software Engineer, Tapestry

Google • Mountain View, CA • 6h ago

IT & Engineering

Fr on tE nd So ft wa re En gi ne er ,T ap es tr y Software Engineering Mountain View, CA (HQ) About the team: We are a team working to rapidly decarbonize the electric grid....

6h ago

Go to Apply

Data Engineer- Maps

Apple • Cupertino, CA • 6h ago

IT & Engineering

Data Engineer- Maps Cupertino, California, United States Software and Services Summary Posted: May 05, 2025 Weekly Hours: 40 Role Number: 200602790 Apple is where individual imaginations gather together, c...

6h ago

Go to Apply

Subscribe to get new jobs sent directly to you
(no charge, cancel anytime)

This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.