Senior AI Accelerator Architect - DL Systems (Equity)

NVIDIA Corporation; Santa Clara (CA); On-site; USD 224,000 - 357,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Corporation in Santa Clara, CA is seeking a Senior Deep Learning Systems Architect to design hardware accelerator and processor architectures for state-of-the-art ML and data analytics on GPUs, mobile, embedded and datacenter platforms.

You will analyze DL workloads, co-design systems, benchmark ideas, and collaborate with DL researchers, hardware architects, and software engineers to advance AI compute performance.

Qualifications

• MS or PhD in CS, CS architecture, EE, or related field with 10+ years of experience.
• Strong ML/DNN background with DL fundamentals.
• Experience with training DNNs using PyTorch, TensorFlow or JAX.
• Proficient in C++ and Python for system design.

Responsibilities

• Design hardware accelerator and processor architectures for AI workloads.
• Contribute to features enabling AI systems on GPUs, mobiles, and data centers.
• Co-design systems from conception to prototyping and benchmarking.
• Analyze DL workloads and map to hardware with bottleneck fixes.
• Collaborate with DL researchers, hardware architects, and software engineers.

Skills

Deep Learning

C++

Python

GPU Computing

HPC MPI/OpenMP

DL Frameworks

Education

MS or PhD in Computer Science / CS Architecture / Electrical Engineering

Tools

PyTorch

TensorFlow

JAX

CUDA

OpenCL

OpenACC

MPI

OpenMP

Job description

NVIDIA Corporation in Santa Clara, CA is seeking a Senior Deep Learning Systems Architect to design hardware accelerator and processor architectures for state-of-the-art ML and data analytics on GPUs, mobile, embedded and datacenter platforms.

You will analyze DL workloads, co-design systems, benchmark ideas, and collaborate with DL researchers, hardware architects, and software engineers to advance AI compute performance.