英伟达Deep Learning Compiler Verification and Infra Development Intern - 2026
任职要求
• Pursuing BS/MS/PhD in Computer Science, Computer/Electrical Engineering, Mathematics or equivalent program. • Strong Python programming skills • Strong modern C++ programming skills • Good understanding of machine learning domain and concepts, includ…
工作职责
Do you want to help drive the progress of compilers for machine learning applications? Are you excited to learn how GPU performance powers technology such as mobile gaming, machine learning (ML), and self-driving cars? Are you passionate about challenging yourself and would you love to contribute as part of a world-class company? We are building the next generation of compiler technologies to accelerate ML workloads. We are looking for an intern to work on compiler verification & testing. What you’ll be doing: In this role you will work closely with deep learning compiler developers to verify new and state of the art deep learning related features and components including implementing and executing functional and performance testing and benchmarking software solutions. This would include implementing verification programs, tools, scripts, and libraries. You will apply deep learning and other sophisticated techniques to implement compiler verification solutions.
• Design and implement the DSL and the core compiler of tile-aware GPU programming model for emerging GPU architectures • Continuously innovate and iterate on the core architecture of the compiler to consistently optimize performance • Investigation of next-generation GPU architectures and provide solutions in the DSL and compiler stack • Performance analysis on emerging AI/LLM workloads and integrate with AI/ML frameworks
We are now looking for a Deep Learning Performance Software Engineer! We are expanding our research and development for deep learning. 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. 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 deep learning compiler • Develop highly optimized deep learning kernels • End-to-end performance optimization • Do performance optimization, analysis, and tuning