Software Engineer, Deep Learning – Autonomous Vehicles
Job Description
We are looking for outstanding Deep Learning Software Engineers to develop and productize NVIDIA's deep learning solutions in autonomous driving vehicles. We work on innovative vision and language models to ensure they are accelerated by NVIDIA SW. In addition, we contribute to our open-source library and have opportunities to present our work at conferences. As a member of our Solution Engineering-Automotive Deep Learning team, you will apply ground breaking NVIDIA deep learning model training/inference software libraries for deployment on NVIDIA's hardware architecture. You will develop new deep learning architectures, train deep learning models, and compile and optimize DNN graphs. As a part of this role, you will be building a close technical relationship with our automotive partners during product development and coordinate with the architecture and software teams to develop the best solution for partners working on our platforms.
What you'll be doing:
Train, fine-tune, optimize and customize perception DNNs in low precision (FP16/INT8)
Apply various quantization methods of DNNs
Improve DNN architectures using ML algorithms on NVIDIA GPUs or DLAs
Continuously improve inference speed, accuracy and power consumption of DNNs
What we need to see:
A MS or PhD in Computer Architecture, Computer Science, Electrical Engineering or related field or equivalent experience
2+ years of work experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc.)
Strong Python and C/C++ programming skills
Ways to stand out from the crowd:
Experience with NVIDIA software libraries such as CUDA and TensorRT
Familiarity with sophisticated C++11/C++14 language features and CUDA kernel programming.
Open source project ownership or contribution, balanced GitHub repositories, guiding and/or mentoring experience.
NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most hard-working and talented people in the world working for us. If you're creative and passionate about developing cloud services we want to hear from you!
The base salary range is 148,000 USD - 276,000 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.