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
Tech Lead Manager, ML Training & Inference Infra employer: Rivian Automotive
Rivian is an exceptional employer that fosters a culture of innovation and adventure, making it an ideal place for those passionate about autonomous driving and environmental sustainability. Located in a dynamic environment, employees benefit from opportunities for professional growth, mentorship, and collaboration with diverse teams dedicated to pushing the boundaries of technology. With a commitment to inclusivity and employee well-being, Rivian ensures that every team member can thrive while contributing to a mission that protects the planet for future generations.
