Senior ML Engineer for Large-Scale Autonomous Driving AI

Rivian; New York (NY); Sur place; USD 170 000 - 250 000; Plein temps

Résumé du poste

Rivian is seeking a full-time Machine Learning Engineer to build state-of-the-art AI infrastructure for training very large foundation models and accelerating model training and inference. You will join a multidisciplinary team of software engineers, ML engineers, and researchers to push the boundaries of end-to-end autonomous driving solutions.

The role focuses on designing and deploying large DL models, scaling training pipelines, and optimizing performance for our fleet of electric vehicles.

Qualifications

• PhD in CS/CE/EE or equivalent, in industry experience.
• Deep knowledge of PyTorch and transformer architectures.
• Experience with large-scale distributed training and model profiling.

Responsabilités

• Design, train, and deploy large deep learning models for a fleet of vehicles.
• Improve training infrastructure and deployment pipelines to accelerate iteration and performance.

Connaissances

PyTorch

CUDA

Triton

Transformer models

Distributed training

Formation

PhD in CS/CE/EE or equivalent industry experience

Outils

PyTorch Lightning

TensorRT

Description du poste

Rivian is seeking a full-time Machine Learning Engineer to build state-of-the-art AI infrastructure for training very large foundation models and accelerating model training and inference. You will join a multidisciplinary team of software engineers, ML engineers, and researchers to push the boundaries of end-to-end autonomous driving solutions.

The role focuses on designing and deploying large DL models, scaling training pipelines, and optimizing performance for our fleet of electric vehicles.