Senior AI Networking Engineer for ML-Optimized Data Center

NVIDIA; Santa Clara (CA); On-site; USD 152,000 - 241,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a Senior Software Engineer to join the AI Networking team in Santa Clara, California. This role involves developing machine learning tools for optimizing AI workloads across GPU and CPU clusters, supporting next-gen large-scale AI systems.

The ideal candidate has a PhD or Master's degree and 4+ years of related experience. Proficiency in machine learning, networking concepts, and programming languages like Python and C++ is essential. Competitive salary offered, with benefits and equity options.

Qualifications

• 4+ years of experience applying machine learning techniques to optimization problems.
• Experience in high-performance computing (HPC), networking, and AI applications.
• Hands-on development of various learning algorithms.

Responsibilities

• Design resource allocation and optimization techniques for LLM models.
• Optimize Deep Learning training on NVIDIA supercomputers.
• Build ML tools for performance prediction with a focus on networking.
• Collaborate with hardware/software teams for performance analysis.

Skills

Machine Learning

Deep Learning

Networking

Python

Bash

C++

Reinforcement Learning

Education

PhD or Master's degree in Computer Science or Software Engineering

Tools

PyTorch

TensorFlow

CUDA

NCCL

Job description

NVIDIA is seeking a Senior Software Engineer to join the AI Networking team in Santa Clara, California. This role involves developing machine learning tools for optimizing AI workloads across GPU and CPU clusters, supporting next-gen large-scale AI systems.

The ideal candidate has a PhD or Master's degree and 4+ years of related experience. Proficiency in machine learning, networking concepts, and programming languages like Python and C++ is essential. Competitive salary offered, with benefits and equity options.