GPU Kernel Development Engineer at Wayve

Join Wayve as a GPU Kernel Development Engineer and drive innovation in autonomous vehicle AI models. Collaborate with teams to optimize performance using cutting-edge GPU technology.
This role based in London requires an individual with over five years of GPU kernel development experience. You will be at the forefront of transforming AI models for deployment on consumer vehicles while mastering GPU architecture like NVIDIA Orin and Qualcomm Snapdragon. Your contributions will significantly enhance inference performance in embedded environments.
Key Responsibilities:
• Optimize model compilation for hardware utilization
• Transform AI models for low precision implementations
• Act as GPU architecture expert for specific platforms
• Lead the model creation process from PyTorch to native code
Requirements:
• 5+ years in GPU development
• Strong skills in C++ programming
• Proficient with CUDA and TensorRT tools
• In-depth knowledge of GPU operations and d...