Wayve

Staff ML Performance Engineer focused on optimizing large-scale ML training and inference workloads for Embodied AI technology in autonomous driving systems. The role involves profiling ML jobs, designing and implementing efficiency improvements (parallelism, model compilation, mixed precision), developing observability tools, and creating benchmarking tools to track performance gains. Collaboration with Research teams is key to integrate efficiency improvements and foster a performance optimization culture.