Rivian is seeking a Tech Lead – Manager – ML training and infrastructure to join the Perception team. You will shape state-of-the-art ML infrastructure for training and inference of large autonomous driving models, driving performance and scalability across GPU clusters.
The role requires deep expertise in transformer-based models, distributed training, and profiling to accelerate training and inference on onboard and edge systems, with a strong emphasis on reliability and maintainability.
#J-18808-Ljbffr…
