Sundayy
ML Engineer, Performance
How your CV stacks up
Upload your CV to see how well it fits this job role
?%
About The Company
Founded in 2017, Wayve is at the forefront of Embodied AI technology development. Our innovative AI software and foundation models empower vehicles to perceive, understand, and navigate complex environments, significantly advancing the safety and usability of automated driving systems. Our mission is to create autonomous solutions that propel the world forward, with a focus on mapless, hardware-agnostic AI products tailored for automakers. This approach accelerates the transition from assisted to fully automated driving, shaping the future of mobility.
At Wayve, we thrive in a dynamic and fast-paced environment where big problems inspire groundbreaking solutions. We embrace uncertainty, challenge conventional thinking, and remain committed to continuous learning and innovation. Our culture values diversity, inclusion, and collaboration, fostering an environment where every contribution matters. We believe in supporting each other to deliver impactful results and making Wayve a place where your career can define your future.
About The Role
As a Staff ML Performance Engineer at Wayve, you will be instrumental in high-impact projects focused on optimizing machine learning inference for edge accelerators and GPUs. Your primary responsibility will be to enable large transformer-based models to operate efficiently on low-cost, low-power edge devices, which is critical for launching Wayve’s first autonomous driving product. This role involves setting the technical direction for transforming these models into reliable, production-ready systems that function seamlessly within in-vehicle compute environments.
Reasons to use Rodeo
Start with a chat, not a search bar
Grad scheme, placement, apprenticeship? Not sure what you want yet — that's fine. Your agent talks it through with you and turns "I have no idea" into a shortlist.
It searches the market for you
Every day your agent scans the market matching roles against what actually matters to you, not just keywords on a CV.
Only hits
No noise. No "maybe this fits." Just roles with a clear explanation of why they're right — and where to focus when applying.
This position is highly hands-on, requiring collaboration across ML systems, compilers, runtimes, kernels, and embedded deployment. You will contribute to several early-stage projects, working closely with cross-functional teams to improve inference performance, develop innovative solutions, and support deployment on various hardware targets. Your expertise will directly influence the scalability, efficiency, and robustness of our autonomous driving technology.
Qualifications
- Proven experience in performance optimization within production systems constrained by latency, memory, bandwidth, power, thermal, or cost.
- Strong proficiency with relevant stacks and toolchains such as TensorRT, CUDA, Qualcomm QNN, Triton, or OpenCL.
- Ability to operate effectively across multiple levels of abstraction, from high-level model behavior to low-level kernel and runtime execution.
- Excellent software engineering fundamentals, including debugging, profiling, testing, and writing maintainable code.
- Effective communication skills and a collaborative mindset, capable of aligning stakeholders on performance trade-offs and priorities.
Responsibilities
- Profile and identify bottlenecks across the full inference stack, including model graph, compiler/runtime, kernel execution, and memory movement, delivering measurable improvements.
- Implement and validate optimizations such as operator fusion, scheduling, quantization-aware performance enhancements, and custom kernels.
- Develop robust benchmarking and regression testing frameworks to ensure consistent performance improvements across models, devices, and software releases.
- Optimize inference performance for multiple hardware targets, including NVIDIA Orin/Thor and Qualcomm platforms, ensuring maintainability and scalability.
- Collaborate with model developers to influence architecture and training/deployment decisions that impact on-device performance.
- Contribute to the development of technical roadmaps, tooling, and standards to elevate performance engineering practices across the team.
Get help with your application
Your very own career expert that helps elevate your application to the next level.
Benefits
- Competitive salary and comprehensive health benefits.
- Hybrid working model combining office presence in London and remote work flexibility.
- Opportunities for professional growth and development within a pioneering AI company.
- Inclusive and diverse work environment that values innovation, collaboration, and individual contributions.
- Access to cutting-edge technology and involvement in high-impact projects shaping the future of autonomous mobility.
Equal Opportunity
Wayve is committed to fostering an inclusive workplace where diversity is celebrated. We are an equal opportunity employer and do not discriminate based on race, gender, religion, national origin, age, disability, sexual orientation, gender identity, veteran status, or any other protected characteristic. We ensure that all candidates and employees are treated fairly and with respect throughout the hiring and employment process. We welcome applications from individuals of all backgrounds and are dedicated to creating a culture of equity, inclusion, and belonging.
“It took my CV and asked me questions relevant to understanding what kind of jobs to suggest for me. Suggestions were almost perfect. Jobs were exactly what I’ve been looking for.”
Jessica, London
Skills
Location