Senior Deep Learning Software Engineer, Inference

Job Description

NVIDIA is seeking a Senior Software Engineer specializing in Deep Learning Inference to design, build, and optimize GPU-accelerated software for AI applications.

Requirements

• Masters or PhD in Computer Engineering, Computer Science, EECS, or AI
• 5+ years of relevant software development experience
• Excellent C/C++ programming and software design skills
• SW Agile skills and Python experience are helpful
• Prior experience with training, deploying, or optimizing DL models in production is a plus
• Background with performance modeling, profiling, debug, and code optimization
• GPU programming experience with CUDA, OAI TRITON, or CUTLASS

Responsibilities

• Performance optimization, analysis, and tuning of DL models in various domains
• Scaling performance of DL models across different NVIDIA accelerators
• Contributing features and code to NVIDIA's inference libraries
• Working with cross-collaborative teams across frameworks and libraries

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