Staff Software Engineer, ML Training and Inference Infrastructure at Rivian

As a Staff Software Engineer on the Perception team at Rivian, you will be instrumental in developing advanced machine learning algorithms that are critical for the self-driving features of their electric vehicles. This role requires a deep understanding and enthusiasm for establishing a state-of-the-art ML infrastructure, specifically focusing on the training and inference of large autonomous driving models. You will be responsible for optimizing both training and inference performance, ensuring that Rivian's vehicles are at the forefront of AI-driven automotive technology. This position is integral to Rivian's mission of creating emissions-free Electric Adventure Vehicles and attracting innovative thinkers who challenge the status quo. You will contribute to redefining what's possible in sustainable transportation by optimizing the performance of deep learning workloads on NVIDIA GPU systems, minimizing model inference latency, and designing/deploying large deep learning models that effectively leverage vast amounts of data. Your work will directly impact the safety and capabilities of Rivian's vehicles.