AI and ML Infra Software Engineer, GPU Clusters - New College Grad 2026

NVIDIA • Santa Clara, CA • 7h ago

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's 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, 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. Come join the team and see how you can make a lasting impact on the world.

We are currently hiring an AI/ML Infrastructure Software Engineer at NVIDIA to join our Hardware Infrastructure team. As an Engineer, you will play a crucial role in boosting productivity for our researchers through implementing advancements across the entire stack. Your primary responsibility will involve working closely with customers to identify and resolve infrastructure gaps, enabling innovative AI and ML research on GPU Clusters. Together, we can create powerful, efficient, and scalable solutions as we shape the future of AI/ML technology!

What you will be doing:

Collaborate closely with our AI and ML research teams to understand their infrastructure needs and obstacles, translating those observations into actionable improvements.

Monitor and optimize the performance of our infrastructure ensuring high availability, scalability, and efficient resource utilization.

Help define and improve important measures of AI researcher efficiency, ensuring that our actions are in line with measurable results.

Collaborate with diverse teams, including researchers, data engineers, and DevOps professionals, to build a seamless and coordinated AI/ML infrastructure ecosystem.

Stay on top of the latest advancements in AI/ML technologies, frameworks, and effective strategies, and promote their implementation within the company.

What we need to see:

BS or equivalent experience in Computer Science or related field, with proven experience in AI/ML and HPC workloads and infrastructure.

Hands-on experience in using or operating High Performance Computing (HPC) grade infrastructure as well as in-depth knowledge of accelerated computing (e.g., GPU, custom silicon), storage (e.g., Lustre, GPFS, BeeGFS), scheduling & orchestration (e.g., Slurm, Kubernetes, LSF), high-speed networking (e.g., Infiniband, RoCE, Amazon EFA), and containers technologies (Docker, Enroot).

Expertise in running and optimizing large-scale distributed training workloads using PyTorch (DDP, FSDP), NeMo, or JAX. Also, possess a deep understanding of AI/ML workflows, encompassing data processing, model training, and inference pipelines.

Proficiency in programming & scripting languages such as Python, Go, Bash, as well as familiarity with cloud computing platforms (e.g., AWS, GCP, Azure) in addition to experience with parallel computing frameworks and paradigms.

Passion for continual learning and keeping abreast of new technologies and effective approaches in the AI/ML infrastructure field.

Excellent communication and collaboration skills, with the ability to work effectively with teams and individuals of different backgrounds.

NVIDIA provides competitive salaries and a comprehensive benefits package. Our engineering teams are expanding rapidly due to exceptional growth. If you're a passionate and independent engineer with a love for technology, we want to hear from you.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 120,000 USD - 189,750 USD for Level 2, and 148,000 USD - 235,750 USD for Level 3.

You will also be eligible for equity and benefits (https://www.nvidia.com/en-us/benefits/) .

Applications for this job will be accepted at least until October 17, 2025.

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

ML Data Scientist - Marketplace, Apple Ads

Apple • Cupertino, CA • 1h ago

IT & Engineering

ML Data Scientist - Marketplace, Apple Ads Cupertino, California, United States Software and Services Summary Posted: Oct 13, 2025 Role Number: 200625825-0836 At Apple, we believe in the power of technology to enrich peo...

1h ago

Go to Apply

Camera Machine Learning - Data Scientist

Apple • Cupertino, CA • 1h ago

IT & Engineering

Camera Machine Learning - Data Scientist Cupertino, California, United States Hardware Summary Posted: Oct 13, 2025 Role Number: 200624376-0836 Apple delivers the most popular cameras in the world. Each product release p...

1h ago

Go to Apply

Principal, Software Engineer, Streaming

Walmart • Sunnyvale, CA • 1h ago

IT & Engineering

Position Summary... What you'll do... What you'll do... Walmart's Global Tech Platform team helps thousands of developers save time and code better, so that millions of Walmart associates can help hundreds of millions of...

1h ago

Go to Apply

Staff, Software Engineer, Streaming and Messaging

Walmart • Sunnyvale, CA • 1h ago

IT & Engineering

1h ago

Go to Apply

Senior, Data Scientist

Walmart • Sunnyvale, CA • 1h ago

IT & Engineering

Position Summary... What you'll do... Data Source Identification Requires knowledge of Functional business domain and scenarios Categories of data and where it is held Business data requirements Database technologies and...

1h ago

Go to Apply

Data Scientist

Accenture • Mountain View, CA • 1h ago

IT & Engineering

Accenture Flex offers you the flexibility of local fixed-duration project-based work powered by Accenture, a leading global professional services company. Accenture is consistently recognized on FORTUNE's 100 Best Compan...

1h ago

Go to Apply

Sr. Data Scientist, Subscriptions

Apple • Cupertino, CA • 1h ago

IT & Engineering

Sr. Data Scientist, Subscriptions Cupertino, California, United States Software and Services Summary Posted: Oct 14, 2025 Role Number: 200625975-0836 We're idealists. Inventors. Forever tinkering with products and proces...

1h ago

Go to Apply

Principal, Data Scientist

Walmart • Sunnyvale, CA • 1h ago

IT & Engineering

Position Summary... What you'll do... Code Development and Testing Requires knowledge of coding languages like SQL Java C Python and others Testing methods such as static dynamic software composition analysis manual pene...

1h ago

Go to Apply

Senior, Data Scientist

Walmart • Sunnyvale, CA • 1h ago

IT & Engineering

Position Summary... What you'll do... "Immigration sponsorship is not available in this role." Our Company If you're thinking 'scale', think bigger and don't stop there. At Walmart Labs, we don't just innovate, we enable...

1h ago

Go to Apply

Machine Learning Engineer, AGIF

Amazon • Sunnyvale, CA • 1h ago

IT & Engineering

Description The Artificial General Intelligence (AGI) team is building industry-leading multimodal systems that will transform how customers interact with AI. We are seeking an SDE II with machine learning expertise to b...

1h 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.