Staff Machine Learning, AI Infrastructure, Autonomy

Job Title: Staff Machine Learning, AI Infrastructure, Autonomy Company: Rivian Job Category: Machine Learning Roles & Responsibilities: As a company, we constantly challenge what’s possible, never simply accepting what has always been done. We reframe old problems, seek new solutions and operate comfortably in areas that are unknown. Our backgrounds are diverse, but our team shares a love of the outdoors and a desire to protect it for future generations. Responsibilities: * Optimize the performance of Deep Learning training workload on NVIDIA GPU systems on a Rivian may share your Candidate Personal Data with (i) internal personnel who have a need to know such information in order to perform their duties, including individuals on our People Team, Finance, Legal, and the team(s) with the position(s) for which you are applying; (ii) Rivian affiliates; and (iii) Rivian’s service providers, including providers of background checks, staffing services, and cloud services. Participation in AI profile matching is entirely voluntary. If you prefer that your profile not be used in this process, you can opt out at any time. Opting out means your profile will be excluded from automated matching and will not be surfaced for additional roles through this system. Your current application remains active and will not be affected in any way. Please note that we are currently not accepting applications from third party application services. Work Location: London, ENG, UK Availability: Full time

Qualifications: * PhD in CS/CE/EE, or equivalent, in industry experience * Deep knowledge of PyTorch, TrensorFlow or Jax * Knowledge of model training framework * In\-depth knowledge of transformer architecture and ways to accelerate the training and * Experience of performing large scale distributed training of models * A track record of profiling models and doing detective work to improve model training and Rivian may collect, use and disclose your personal information or personal data (within the meaning of the applicable data protection laws) when you apply for employment and/or participate in our recruitment processes (“Candidate Personal Data”). This data includes contact, demographic, communications, educational, professional, employment, social media/website, network/device, recruiting system usage/interaction, security and preference information. Rivian may use your Candidate Personal Data for the purposes of (i) tracking interactions with our recruiting system; (ii) carrying out, analyzing and improving our application and recruitment process, including assessing you and your application and conducting employment, background and reference checks; (iii) establishing an employment relationship or entering into an employment contract with you; (iv) complying with our legal, regulatory and corporate governance obligations; (v) recordkeeping; (vi) ensuring network and information security and preventing fraud; and (vii) as otherwise required or permitted by applicable law. Key Skills: Machine Learning, PyTorch, Model Training Salary: Not disclosed