英伟达Deep Learning Compiler Engineer - CUDA
任职要求
• Masters or PhD or equivalent experience in relevant discipline (CE, CS&E, CS, AI) • 2+ years of relevant work experience • Excellent C/C++ programming and software engineering skills, ACM background is a plus • Good fundamental knowledges on computer architecture • Strong ability in abstracting problems and the methodology in resolving problems • Strong compiler backgrounds including MLIR/TVM/Triton/LLVM is desired • Good knowledge of GPU architecture an…
工作职责
• 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
• 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
• 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
• Build, maintain, and improve CI infrastructure that supports development, verification, and release of NVIDIA’s deep learning compiler stacks across GPU and accelerator environments • Improve CI reliability and signal quality by reducing flakes, improving reproducibility, strengthening diagnostics, and making correctness and performance failures easier to understand and act on • Apply automation, AI, and agent-based workflows to reduce manual CI operations, speed up failure triage, and improve developer efficiency • Build reusable and self-service CI platforms that support multiple products, projects, model suites, hardware targets, and software configurations while partnering closely with compiler, infrastructure, and release teams
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.