Deep Learning Compiler Engineer - CUDA

Responsibilities

• 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
• What we need to see
• 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

Languages