Senior Performance Software Engineer, Deep Learning Libraries

Responsibilities

• Writing highly tuned compute kernels to perform core deep learning operations (e.g. matrix multiplies, convolutions, normalizations)
• Following general software engineering best practices including support for regression testing and CI/CD flows
• Collaborating with teams across NVIDIA
• CUDA compiler team on generating optimal assembly code
• Deep learning training and inference performance teams on which layers require optimization
• Hardware and architecture teams on the programming model for new deep learning hardware features
• What we need to see
• Masters or PhD degree or equivalent experience in Computer Science, Computer Engineering, Applied Math, or related field
• 2+ years of relevant industry experience
• Demonstrated strong C++ programming and software design skills, including debugging, performance analysis, and test design

Languages