AI Computing Architect

NVIDIA NVIDIA · Semiconductors · Shanghai, China +1

NVIDIA is seeking an AI Computing Architect to develop innovative architectures for deep learning performance and efficiency, analyze trade-offs using models and simulators, and prototype algorithms. The role requires strong programming skills, computer architecture background, and a foundation in machine learning.

What you'd actually do

  1. Develop innovative architectures to extend the state of the art in deep learning performance and efficiency.
  2. Analyze performance, cost and power trade-offs by developing analytical models, simulators and test suites.
  3. Understand and analyze the interplay of hardware and software architectures on future algorithms, programming models and applications.
  4. Prototype key deep learning and data analytics algorithms and applications.
  5. Actively collaborate with software, product and research teams to guide the direction of deep-learning.

Skills

Required

  • Python
  • C++
  • Computer Architecture
  • Performance Modeling
  • Machine Learning
  • Deep Learning

Nice to have

  • GPU Computing
  • CUDA
  • OpenCL
  • Deep Learning Accelerators
  • Deep Neural Network Training
  • Inference
  • Optimization
  • Pytorch
  • Tensorflow
  • TensorRT
  • Open-source AI compilers
  • OpenAI Triton
  • MLIR
  • TVM
  • XLA

What the JD emphasized

  • BS or higher degree in a relevant technical field (CS, EE, CE, Math, etc.) with 3+ years of work experience.
  • Strong programming skills in Python, C, C++.
  • Strong background in computer architecture.
  • Experience with performance modeling, architecture simulation, profiling, and analysis.
  • Strong foundation in machine learning and deep learning.

Other signals

  • AI computing platforms
  • deep learning performance and efficiency
  • hardware and software architectures
  • deep learning and data analytics algorithms