Senior Deep Learning Software Engineer

This role is based in Santa Clara, CA or Redmond, WA.

Compensation:

$224,000 - $356,500/year

Responsibilities:

• Design a scalable architecture for deep learning inference.
• Develop features in high-level frameworks like PyTorch and JAX.
• Optimize GPU performance and develop custom GPU kernels in CUDA/Triton.
• Collaborate with teams across NVIDIA to enhance deployment solutions.

Requirements:

• Masters, PhD, or equivalent experience in Computer Science, AI, Applied Math, or related field.
• 8+ years of relevant work or research experience in Deep Learning.
• Strong proficiency in Python, PyTorch, and related ML tools.
• Excellent software design skills, including debugging and performance analysis.

Tech stack:

Python, PyTorch, CUDA, Triton.