Key technologies and capabilities for this role
Common questions about this position
The base salary range is 184,000 USD and higher (exact upper limit not fully specified).
Yes, the role is open to remote work, though occasional travel is required for customer visits and conferences.
Candidates need 8+ years as an AI/Software Engineer with Python and/or C++ coding experience using AI libraries and GPUs, expertise in profiling and optimizing model training/inference on GPUs, and experience developing/optimizing GPU kernels for deep learning focusing on GEMM and attention kernels. A BS, MS, or PhD in Computer Science, Engineering, Physics, Mathematics, or equivalent is required, along with strong communication and teamwork skills.
The role involves close collaboration with customers, NVIDIA's software engineering, product, and sales teams, and cross-functional teams including Engineering, Research, Sales, Product, and Marketing. It emphasizes being a self-starter with a passion for growth, continuous learning, and sharing insights in a customer-facing environment.
Stand out with full stack experience from DL frameworks like PyTorch/JAX to low-level CUDA/CUTLASS/cuDNN/NCCL, customer-facing skills with enterprise developers, MLOps knowledge in containers/Kubernetes/data centers, large-scale production data pipelines, and creative problem-solving for complex issues.
Designs GPUs and AI computing solutions
NVIDIA designs and manufactures graphics processing units (GPUs) and system on a chip units (SoCs) for various markets, including gaming, professional visualization, data centers, and automotive. Their products include GPUs tailored for gaming and professional use, as well as platforms for artificial intelligence (AI) and high-performance computing (HPC) that cater to developers, data scientists, and IT administrators. NVIDIA generates revenue through the sale of hardware, software solutions, and cloud-based services, such as NVIDIA CloudXR and NGC, which enhance experiences in AI, machine learning, and computer vision. What sets NVIDIA apart from competitors is its strong focus on research and development, allowing it to maintain a leadership position in a competitive market. The company's goal is to drive innovation and provide advanced solutions that meet the needs of a diverse clientele, including gamers, researchers, and enterprises.