Staff Machine Learning, AI Infrastructure, Autonomy
- Location
- Camden Town, England, United Kingdom
Summary As a Staff Engineer - ML training and infrastructure, you will be a member of the Perception team at Rivian, which develops advanced machine learning algorithms that directly impact safety-critical self-driving features of our category-defining vehicles. We are looking for candidates with deep knowledge … infrastructure for training and inference of large autonomous driving models, and optimizing the training and inference performance. Responsibilities Optimize the performance of Deep Learning training workload on NVIDIA GPU systems on a large scale Optimize the latency of model inference and model pre- and post-processing on onboard ...