NVIDIA Logo

NVIDIA

AI and ML Infra Software Engineer, GPU Clusters

Job Posted 2 Hours Ago Posted 2 Hours Ago
Be an Early Applicant
2 Locations
Senior level
2 Locations
Senior level
As an AI/ML Infrastructure Software Engineer at NVIDIA, you will enhance GPU clusters for AI research by optimizing infrastructure, collaborating with teams, and implementing advancements to improve efficiency and scalability in AI/ML workflows.
The summary above was generated by AI

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 5+ years of 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.

The base salary range is 148,000 USD - 287,500 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.

HQ

NVIDIA Seattle, Washington, USA Office

4545 Roosevelt Way NE 6th Floor, Seattle, Washington, United States, 98105

Similar Jobs

3 Hours From Now
18 Locations
Senior level
Senior level
Greentech • Real Estate • Social Impact • Energy • Industrial • Solar • Renewable Energy
As a SCADA Engineer, you will support and troubleshoot SCADA systems for renewable energy projects, verify equipment performance, enhance system reliability, and lead the integration of new assets. You will also create training materials and collaborate with other engineering functions to resolve technical issues.
3 Hours Ago
Remote
Hybrid
58 Locations
84K-202K Annually
Mid level
84K-202K Annually
Mid level
Artificial Intelligence • Professional Services • Business Intelligence • Consulting • Cybersecurity • Generative AI
As a Senior Associate, you will assist clients by aligning their data strategies to their business needs and migrate them to modern cloud environments. You will leverage technologies like AWS, Azure, and Snowflake, developing data governance strategies and advanced analytics solutions while ensuring successful project execution.
9 Hours Ago
Seattle, WA, USA
238K-434K Annually
Expert/Leader
238K-434K Annually
Expert/Leader
Artificial Intelligence • Information Technology • Natural Language Processing • Software • Business Intelligence • Generative AI
The Senior Principal AI Architect at Qualtrics will lead AI initiatives, utilizing deep knowledge of machine learning to drive innovative solutions. Responsibilities include collaborating with multidisciplinary teams, staying informed on AI advancements, and engaging in strategic discussions with leadership.
Top Skills: Deep LearningMachine LearningMlflowNatural Language ProcessingPyTorchReinforcement LearningSagemakerTensorFlow

What you need to know about the Seattle Tech Scene

Home to tech titans like Microsoft and Amazon, Seattle punches far above its weight in innovation. But its surrounding mountains, sprinkled with world-famous hiking trails and climbing routes, make the city a destination for outdoorsy types as well. Established as a logging town before shifting to shipbuilding and logistics, the Emerald City is now known for its contributions to aerospace, software, biotech and cloud computing. And its status as a thriving tech ecosystem is attracting out-of-town companies looking to establish new tech and engineering hubs.

Key Facts About Seattle Tech

  • Number of Tech Workers: 287,000; 13% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Amazon, Microsoft, Meta, Google
  • Key Industries: Artificial intelligence, cloud computing, software, biotechnology, game development
  • Funding Landscape: $3.1 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Madrona, Fuse, Tola, Maveron
  • Research Centers and Universities: University of Washington, Seattle University, Seattle Pacific University, Allen Institute for Brain Science, Bill & Melinda Gates Foundation, Seattle Children’s Research Institute
By clicking Apply you agree to share your profile information with the hiring company.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account