Senior Machine Learning Engineer, AI Infrastructure, Autonomy
On-siteLondon, England, United Kingdom
London, England, United KingdomOn-siteFull TimeSenior LevelDoctorate Or Professional DegreeEnterprise
Full TimeSenior LevelDoctorate Or Professional DegreeEnterprise
Job Summary
Design, train, and deploy large deep learning models leveraging labeled and unlabeled data from a fleet of million vehicles. Improve ecosystem for training infrastructure and deployment pipelines to accelerate model iteration and performance. Profile models and conduct detective work to optimize training and inference speed. Requires PhD in CS/CE/EE, deep PyTorch knowledge, and experience with Cuda or Triton. Previous autonomous driving experience and Nvidia TensorRT familiarity are preferred. Rivian challenges existing solutions to protect the outdoors for future generations.
Required Qualifications
- PhD in CS/CE/EE, or equivalent, in industry experience
- Deep knowledge of PyTorch
- Experience with Cuda or Triton language for writing custom ops
- Knowledge of model training framework (e.g. PyTorch Lightning)
- In-depth knowledge of transformer architecture and ways to accelerate the training and inference of transformer models
- Experience of performing large scale distributed training of models
- A track record of profiling model and doing detective work to improve model training and inference speed
Desired Qualifications
- Previous experience in the autonomous driving industry
- Knowledge of Nvidia TensorRT
- Experience with edge computing systems
- Knowledge of model optimization including quantization, pruning, etc.
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.