Senior Deep Learning Compiler Engineer

Job Summary

A company is looking for a Senior Deep Learning Compiler Engineer - CUDA.

Key Responsibilities

• Design and implement Pythonic language interface for tile-aware GPU programming
• Optimize compiler pipelines for efficient execution and integrate with AI/ML frameworks
• Collaborate with hardware teams to co-design compiler optimizations for emerging GPU architectures

Required Qualifications

• Bachelor's degree in Computer Science, Electrical Engineering, or related field; MS or PhD preferred
• 5+ years of experience with ML/DL systems development, preferably for compilers
• Strong programming skills in Python and C/C++
• Expert experience in developing or using deep learning frameworks (e.g., PyTorch, JAX, Triton)
• Proficiency in compiler frameworks and a deep understanding of GPU architectures