AI Network System Architect

rs worldwide will use.

What You'll Be Doing:

• Investigating emerging technologies and methodologies in ML and AI to discern their interactions with network infrastructure.

• Executing workloads on AI systems, conducting profiling, and analyzing bottlenecks and possible enhancements.

• Conducting research and implementing optimizations for communication libraries like NCCL and UCX.

• Spearheading the conceptualization of next-generation networking products tailored to support and accelerate state-of-the-art ML workloads.

• Develop models for simulations, analyze simulation results, and develop optimization algorithms.

• Collaborate with multi-functional teams, including other architecture teams, logic design, system software, firmware, and ML research teams, to ensure the successful execution of the project.

What We Need To See:

• M.Sc, or Ph. D degree in Computer Science, Computer Engineering, or Electrical Engineering.

• At least 2+ years of industry or research experience in computer networks.

• Extensive expertise in ML/AI workloads, particularly in distributed training.

• Excellent understanding of large-scale network behavior and the effect of distributed computing workloads on the network.

• Experience in the development of simulation environments.

• Great problem-solving and critical-thinking skills.

• Ability to thrive in a fast-paced and dynamic environment is necessary.

• Work concurrently with multiple groups in the organization.

Ways To Stand Out Of The Crowd:

• Knowledge of communication libraries such as NCCL, UCX, and UCC.

• Good knowledge of network protocols - such as InfiniBand, IP, TCP, RoCE, and network topologies.

• Experience with Python, C++, and dockers.

• Expertise in system engineering, operations research, and intricate hardware-software integrated systems.

• Demonstrated experience in DLRM, LLM or other generative AI.

NVIDIA has some of the most forward-thinking and hardworking people in the world working for us, and due to unprecedented growth, our world-class engineering teams are growing fast. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you.

We are committed to fostering a diverse work environment and are 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. We will ensure that individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, perform essential job functions, and receive other benefits and privileges of employment. Please contact us to request accommodation.