Develop and optimize the systems powering large-scale distributed machine learning training. Responsibilities include profiling performance with Nsight and PyTorch Profiler, optimizing data loading and communication, developing GPU kernels in Triton or CUDA, and improving training throughput, reliability, and cost using PyTorch Distributed.

The candidate will define and execute motion planning projects, develop behavior models using ML techniques, and collaborate with teams to enhance autonomous vehicle performance.