Senior Software Engineer, AV Infra

Location: China, Shanghai

Time Type: Full time

Job Description

We are now looking for Senior Software Engineer of AV Infrastructure and MLOps team! NVIDIA is playing a significant role in AI and Autonomous Vehicles industry with its powerful GPU products, high-performance DRIVE platform and a large eco-system of partners. We collaborate with multiple teams and focus on E2E ground truth data pipeline for AV perception DL models, including data mining, AV data pre-processing and post-processing pipelines. Our work will be fun and very exciting, and have a huge impact in cars worldwide.

What you’ll be doing: 

• Will mainly focus on building automatic pipelines of GT generation for AV DL models of perception, plan and control, including data mining, data processing, auto-labeling, ground truth generation and diagnosis, and performance optimization of the pipelines running on cloud service. We need to have passion for both deep learning and autonomous driving domain! 

• Work closely with AV perception core team, AV production team, AI infrastructure and MLOps engineers to accelerate and optimize the AV perception DNN development and deployment, and achieve the high quality and reliability of AV perception in NVIDIA Drive products. 

What we want to see: 

• MS or PHD candidate in Computer Science, Computer Engineering, Electrical Engineering, Robotics, or related majors 

• Excellent programming, software design, and testing skills in Python, C++, CUDA, or similar languages. 

• 5+ years of experience in either of the following areas:

• 3D Computer Vision, Self-Driving-Cars, Robotics, SLAM, Deep Learning, GPU technology. 

• Good understanding of system architecture, CPU/GPU/Memory/Storage, everything related to performance optimization. 

• Solid knowledge of Linux programming and should be familiar with working under Linux environments. 

• Fluent in both oral and written English. 

Ways to stand out from the crowd: 

• Background in computer vision, machine learning, deep learning, statistics 

• Knowledge of LLM optimization, acceleration and application. Experience with Nvidia Triton will be a big bonus 

• Knowledge of object detection/tracking, multi-sensor fusion such as lidar or camera, time synchronization, and OpenCV/OpenGL programming. 

• Experience with large data exploration, analysis and visualization with Python and SQL. 

• Experience with deep learning models training with popular frameworks like Tensorflow, Pytorch, Caffe and model deployment. 

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We are an equal opportunity employer and value diversity at our company. We have the most talented people in the world working for us. Will you be a creative and autonomous software engineer with a genuine passion for advancing the state of AI and deep learning across a variety of verticals? Looking forward to have you join us!