Deep Learning Performance Architect
At a glance
Mid-level AI Research role at NVIDIA. China, Shanghai · full-time.
Pay not stated AI Research salary
aimlroles summary, based on the employer's posting.
What you'll do
- Build highly optimized deep learning kernels for inference workloads
- Analyze and tune performance, using profiling, debugging and optimization work
- Coordinate with automotive, image-understanding and speech teams to create technical solutions
- Travel occasionally to conferences and customer sites for technical advice and training
- Help implement recent algorithms for public release in TensorRT
What you bring
- At least five years of relevant work experience
- A master’s degree, doctorate or equivalent experience in a related discipline such as computer science or AI
- Strong C and C++ programming ability, paired with solid software design skills
- GPU programming experience with CUDA or OpenCL is desired
Who this fits
This role suits a software engineer focused on deep-learning inference performance and GPU programming. You’ll work across teams and should be comfortable in a fast-paced, customer-focused environment with strong communication. Occasional conference and customer travel is part of the role.
From the employer
We are now looking for a Deep Learning Performance Software Engineer! We are expanding our research and development for Inference. We seek excellent Software Engineers and Senior Software Engineers to join our team.
We specialize in developing GPU-accelerated Deep learning software. Researchers around the world are using NVIDIA GPUs to power a revolution in deep learning, enabling breakthroughs in numerous areas. Join the team that builds software to enable new solutions. Collaborate with the deep learning community to implement the latest algorithms for public release in Tensor-RT. Your ability to work in a fast-paced customer-oriented team is required and excellent communication skills are necessary.
What you’ll be doing:
- Develop highly optimized deep learning kernels for inference
- Do performance optimization, analysis, and tuning
- Work with cross-collaborative teams across automotive, image understanding, and speech understanding to develop innovative solutions
- Occasionally travel to conferences and customers for technical consultation and training
What we need to see:
- Masters or PhD or equivalent experience in relevant discipline (CE, CS&E, CS, AI)
- SW Agile skills helpful
- Excellent C/C++ programming and software design skills
- Python experience a plus
- Performance modelling, profiling, debug, and code optimization or architectural knowledge of CPU and GPU
- GPU programming experience (CUDA or OpenCL) desired
- 5 years of relevant work experience
NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most brilliant and talented people on the planet working for us. If you're creative and autonomous, we want to hear from you!