Senior Deep Learning Software Engineer - Autonomous Vehicles

NVIDIA NVIDIA · Semiconductors · Santa Clara, CA +3 · Remote

Senior Deep Learning Software Engineer focused on developing and productizing deep learning solutions for autonomous vehicles. The role involves training, fine-tuning, optimizing perception DNNs, applying quantization, improving DNN architectures, and enhancing inference speed and power consumption. It requires strong programming skills, experience with deep learning frameworks, computer vision tasks, and familiarity with CNNs and Transformer architectures. Experience with low precision inference, quantization, and NVIDIA software libraries is a plus.

What you'd actually do

  1. Train, fine-tune, optimize and customize perception DNNs in low precision (FP16/INT8)
  2. Apply sophisticated quantization of DNNs
  3. Improve DNN architectures using ML algorithms on NVIDIA GPUs or DLAs
  4. Continuously improve inference speed, accuracy and power consumption of DNNs
  5. Stay up to date with the latest research and innovations in deep learning, implement and experiment with new insights to improve NVIDIA's automotive DNNs.

Skills

Required

  • MS or PhD degree in computer science, computer vision, computer architecture or equivalent experience in technical field
  • 5+ years of work experience in software development
  • 2+ years of experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc.)
  • Experience with solving a computer vision task using deep neural networks, such as object detection, scene parsing, image segmentation.
  • Strong Python and/or C/C++ programming skills
  • Proven technical foundation in CPU and GPU architectures, containers (nvidia-docker), numeric libraries, modular software design
  • Familiar with CNNs and Transformer architectures

Nice to have

  • Experience with low precision inference, quantization, compression of DNNs
  • Experience with NVIDIA software libraries such as CUDA and TensorRT
  • Open source project ownership or contribution, healthy GitHub repositories, guiding and/or mentoring experience

What the JD emphasized

  • 5+ years of work experience in software development
  • 2+ years of experience in developing or using deep learning frameworks
  • Experience with solving a computer vision task using deep neural networks
  • Proven technical foundation in CPU and GPU architectures, containers (nvidia-docker), numeric libraries, modular software design

Other signals

  • Develop and productize NVIDIA's deep learning solutions in autonomous driving vehicles
  • Apply ground breaking NVIDIA deep learning model training/inference software libraries
  • Develop new deep learning architectures, train deep learning models, and compile and optimize DNN graphs
  • Coordinate with the architecture and software teams to develop the best solution for partners