wayve - Senior ML Performance Engineer, Inference Optimisation
Requirements
• Experience deploying or benchmarking ML models on embedded or edge devices • Familiarity with NVIDIA and/or Qualcomm SoCs and performance tooling • Python and C++ proficiency • Experience supporting other engineers or contributing to technical direction within a small team • 🌱 Not ticking every box? That’s totally okay! If you’re passionate about autonomy and keen to learn, we encourage you to apply even if you don’t meet every requirement.
Responsibilities
• This is a hands-on role across ML systems, compilers, runtimes, kernels and embedded deployment. You'll find bottlenecks, implement performance improvements, and work with model developers on performance trade-offs so deployment-aware decisions are made early. • 🧩 What you'll be working on • Profiling inference performance across model graphs, compiler/runtime behaviour, kernel execution and memory movement • Implementing and validating optimisations in compilers, runtimes and/or kernels, including fusion, scheduling, quantisation-aware performance and custom kernels • Building benchmarking and regression tests to track performance across models, devices and software releases • Optimising for edge targets such as NVIDIA Orin/Thor and Qualcomm platforms • Contributing to team tooling, documentation and technical discussions around ML performance • 🙌 You should apply if • You have experience improving performance in production or production-adjacent systems with latency, memory, bandwidth, power, thermal or cost constraints • You have strong hands-on experience with at least one relevant stack, such as TensorRT, CUDA, Qualcomm QNN, Triton or OpenCL • You are comfortable working from high-level model behaviour down to kernel/runtime-level execution • You have strong software engineering fundamentals, including debugging, profiling, testing and maintainable code • You communicate clearly and work well across ML, systems and deployment teams
Benefits
• 💰 Salaries benchmarked against the market annually • 📈 Meaningful equity, sharing in the ownership and long term success of Wayve • ✈️ Relocation support and visa sponsorship where applicable • ✅ Hybrid working, core hours and the chance to work hands on in vehicle workshops and labs • 📚 Learning and development budgets with support for training, conferences and growth • 🩺 Comprehensive benefits including health insurance, dental, enhanced maternity and paternity leave, retirement or pension where applicable, access to therapists, wellbeing partnerships, team socials and more • Wayve is not a mature, fully-structured place with the playbook already written. Much of how we work is still being written, and if you join, you’ll help write it. That suits people who want real ownership more than people who need a settled structure from day one. • If that sounds like the kind of problem you want to spend your time on, we’d really like to hear from you. • At Wayve we're committed to creating a diverse, fair and respectful culture that is inclusive of everyone based on their unique skills and perspectives, and regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, veteran status, pregnancy or related condition (including breastfeeding) or any other basis as protected by applicable law. • For more information visit Careers at Wayve. https://wayve.ai/careers/ To learn more about what drives us, visit Values at Wayve https://wayve.ai/careers/
Apply in one click
Upload My Resume
Drop here or click to browse · Tap to choose · PDF, DOCX, DOC, RTF, TXT