AI Compiler Engineer for Startup

Join a forward-thinking startup as an AI Compiler Engineer focused on optimizing deep learning training. Leverage your expertise in C++, Python, and performance engineering for innovative compiler development.
This role offers the chance to work closely with a skilled team at the forefront of AI compiler technology. You will build and optimize essential compiler components for training workloads while developing passes and hardware backends. This position demands strong technical skills in CUDA, LLVM, and experience tuning for GPUs.
Key Responsibilities:
• Build and optimize AI compiler components for training workloads
• Develop compiler passes and hardware backends efficiently
• Apply kernel-level performance optimizations and enhancements
• Collaborate with engineers focused on ML and LLM
• Fine-tune performance on GPUs or hardware accelerators
Requirements:
• Proficient in Python and C/C++ programming
• Experience with CUDA and LLVM frameworks
• Fam...