Senior Performance Engineer - Deep Learning

Job Description

NVIDIA is seeking senior software engineers to develop and optimize tools for Deep Learning engineers worldwide to design, develop, and deploy AI applications.

Requirements

• BS in Computer Science, Electrical Engineering, or related field (or equivalent experience)
• 6+ years of C++ and Python programming experience
• Strong background in parallel programming, preferably on GPUs
• Knowledge of Computer Architecture and/or Operating Systems
• Proven experience in developing large software projects
• Excellent verbal and written communication skills

Responsibilities

• Develop and optimize open-source libraries like Transformer Engine and TensorFlow Distributed Embeddings
• Study and tune Deep Learning training workloads at large scale
• Build and support NVIDIA submissions to community benchmarks like MLPerf
• Optimize performance of influential Deep Learning models for NVIDIA GPUs
• Explore new technologies and advise on new hardware generations and core platform software components

Benefits

• Multiple relocation packages
• Two weeklong shutdowns (mid-summer and year-end) in the US (in addition to PTO)
• 8-week parental leave
• 9 Employee Resource Groups
• Annual bonus offering
• Flexible work arrangements
• Up to 6% 401K matching