Improving communication for Physical AI-related distributed transformer workloads by employing communication tools developed by NVIDIA. These include NCCL, NCCL GIN, and NVSHMEM, as well as open-source solutions like DeepEP and NCCL EP. This demands in-depth study of interconnect topologies (NVLINK) and network protocols (InfiniBand/RoCE) to develop efficient data transfer strategies alongside techniques enabling compute-communication overlap.
What we need to see
BSc/MSc/PhD or equivalent experience in Computer Science, Electrical Engineering, Physics, Mathematics, or a related technical field.
8+ years of hands-on validated ML/DL performance engineering experience with focus on improving GPU compute efficiency of training and inferencing workloads.
Experience with C, C++, or Python and proficiency with Linux.
Solid understanding of software development, programming techniques, and algorithms.
Strong mathematical fundamentals, including linear algebra and numerical methods.
Background in parallel programming and high-performance computing, with extensive knowledge of parallel architectures and methods for performance analysis and tuning. Experience in GPU programming is desirable.
Experience in distributed communication optimization is highly helpful. This involves familiarity with remote direct memory access, GPU interconnects, collective communication algorithms, and associated open-source libraries used in large-scale model training and inference.
Effective verbal/written communication, and technical presentation skills. Ability to communicate your ideas/code clearly through blog posts, GitHub, ppt.
Ways to stand out from the crowd
Prior experience in writing CUDA kernels, and experience with Nsight System and Nsight Compute.
Experience in comprehensive evaluation and improvement of full-stack systems within at least one of these areas: LLM and HPC. Having expertise ranging from operator-level through framework-level to algorithm-level optimization is strongly preferred.
Proven software engineering fundamentals and system architecture thinking, with the ability to build modules and lead engineering approaches in complex systems.
Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com/
#NALASAHiring
Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.
You will also be eligible for equity and benefits.
Applications for this job will be accepted at least until July 21, 2026.
This posting is for an existing vacancy.
NVIDIA uses AI tools in its recruiting processes.
NVIDIA is committed to fostering an inclusive 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.
