Rivian is seeking a full-time Senior ML, AI Infra Engineer to architect and optimize the AI infrastructure for training very large foundation models and accelerating inference.
You will collaborate with software engineers, ML engineers and researchers to push state-of-art ML for autonomous driving, designing scalable training pipelines and robust deployment.
Applicants should have PhD, PyTorch expertise, CUDA/Triton experience, and proven distributed training capabilities.
#J-18808-Ljbffr
Senior ML Infra Engineer for Autonomous Driving 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.
