This role is for an experienced software professional focused on optimizing deep learning systems using CUDA. The team is dedicated to exploring innovative ideas that enhance the performance of AI workloads on modern computing architectures. You will work on model optimization, custom kernel development, and distributed computing systems. The position requires a strong background in deep learning and systems programming to push the boundaries of AI technology.

• BS, MS, or PhD degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
• 8+ years of relevant industry experience or equivalent academic experience after degree achievement.
• Strong proficiency in C++ and Python programming.
• Solid background in the fundamentals of Deep Learning with a focus on transformers.
• Strong understanding of distributed computing principles and multi-node scaling.
• Proven experience in systems programming and low-level systems performance optimization.
• Familiarity with deep learning accelerator architectures and hands-on experience with CUDA programming.
• Experience profiling and optimizing innovative vision models or generative AI architectures.

• Explore, research, and prototype novel systems optimizations for advanced deep learning models.
• Architect and optimize distributed computing systems that scale from a single node to cluster-scale environments.
• Design, implement, and optimize custom high-performance CUDA kernels for neural network architectures.
• Analyze hardware-software interactions to identify and resolve performance bottlenecks.
• Collaborate with AI researchers and system architects to co-design systems and algorithms.
• Develop exploratory tools and runtime systems to profile and accelerate deep learning paradigms.
• Write clean, effective, and maintainable code for exploratory prototypes.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.