Nvidia Senior Deep Learning Computer Architect 

United States, Texas 

355604202

What you'll be doing:

• As a member of our deep learning architecture team, you will contribute to features that help next-generation GPUs advance the state of AI.

• This position requires you to keep up with the latest DL research and collaborate with diverse teams (internal and external to NVIDIA), including DL researchers, hardware architects, and software engineers.

• Your day to day work will include analyzing the behavior of various deep learning methods, proposing new features to accelerate or enable various methods, and studying the benefits of the proposed features.

What we need to see:

• MS or PhD degree in computer science, computer architecture, electrical engineering or related field or equivalent experience.

• 5+ years of relevant experience in at least a few of the following relevant areas is required in your work history:

  • Computer architecture, including GPU and system level architecture;

  • Performance analysis and optimization;

  • Experience withLLMworkloads, including performance tuning considerations such as parallelization and fusion strategies;`

  • Experience with core deep learning kernels such as matrix multiply, attention, and communicationconvolution

• Programming fluency with C++ and ideally Python

• Experience with GPU computing (CUDA)

• Experience with deep learning frameworks like PyTorch