Senior Deep Learning Systems Architect

About the role

NVIDIA is seeking architects to help design hardware accelerator and processor architectures that enable state-of-the-art machine learning and data analytics algorithms and applications on our next-generation mobile, embedded, and datacenter platforms. This position offers the opportunity to have a real impact in a dynamic, technology-focused company.

Responsibilities

• Contribute to features that help next-generation GPUs and systems advance the state of AI.
• Keep up with the latest deep learning research and collaborate with diverse teams (internal and external to NVIDIA), including DL researchers, hardware architects, and software engineers.
• Participate in engineering projects and co-design architecture for systems from conception, specification, and prototyping.
• Understand various AI/DL workloads and their mapping to underlying hardware and systems.
• Identify potential improvements and bottlenecks, propose solutions to address existing gaps, and accelerate/improve current systems/methods.
• Perform comprehensive analyses from first principles of various deep learning techniques and system optimizations to build analytical models, implement prototypes, and benchmark to test/prove ideas.

Requirements

• MS (or equivalent experience) or PhD degree in computer science, computer architecture, electrical engineering, or related field with 10+ years of relevant work experience. Additional equivalent experience in several of the relevant areas listed below can substitute for an advanced degree.
• Strong background in at least a few of the following relevant areas: machine learning (with focus on Deep Neural Networks), including a solid understanding of DL fundamentals; experience adapting and training DNNs for various tasks; experience developing code for one or more of the DNN training frameworks (such as PyTorch, TensorFlow, or JAX); numerical analysis; performance analysis and optimization; computer architecture.
• Programming fluency with C++ and ideally Python.
• Work experience with GPU computing (CUDA, OpenCL, OpenACC) and HPC (MPI, OpenMP) is a plus.

Pay

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 224,000 USD - 356,500 USD. You will also be eligible for equity and benefits.

JR2003682