AI Engineer
Develop Health
Python
Software Engineer – Deep Learning Performance
NVIDIA
full-time
Posted on:
Origin: • 🇨🇳 China
Visit company websiteJob Level
Mid-LevelSenior
Tech Stack
Python
About the role
- Develop deep learning compiler
- Develop highly optimized deep learning kernels
- End-to-end performance optimization
- Do performance optimization, analysis, and tuning
- Work in a fast-paced customer-oriented team and communicate with stakeholders
Requirements
- 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
- MLIR experience a plus
- AI agent experience a plus
- Performance modelling, profiling, debug, and code optimization or architectural knowledge of CPU and GPU
- GPU programming experience (CUDA or OpenCL) desired
- 3 years of relevant work experience
- Ability to work in a fast-paced customer-oriented team
- Excellent communication skills
