Senior Deep Learning Kernel Software Performance...
Role details
Job location
Tech stack
Job description
-
Analyze, visualize, and optimize software performance using analytical models, simulators, and test suites.
-
Collaborate closely across NVIDIA teams such as:
-
CUDA Compiler teams to identify performance issues.
-
AI/ML training and inference performance teams to identify and optimize critical deep learning layers.
-
hardware architecture performance teams to define expectation for emerging deep learning hardware features.
Requirements
-
A Master's or PhD in Computer Science, Electrical Engineering or Computer Engineering, or equivalent experience.
-
5+ years of relevant industry or research experience.
-
A strong foundation in machine learning and deep learning fundamentals to complement your expertise in computer architecture.
-
A strong background in high performance kernel (such as CUTLASS), work experience on math library performance analysis and profiling to identify performance bottlenecks.
-
Fluency in programming languages such as Python, C, C++.
-
Experience and familiarity with GPU computing and parallel programming models.
-
You have firsthand work experience with analytical performance modeling, profiling, and analysis.
Benefits & conditions
Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 218,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4.