Job Description
As a Senior Deep Learning Engineer – Frameworks, you will be a member of the Deep Learning team at Rivian, which develops advanced machine learning algorithms that directly impact safety and self-driving features of our category defining vehicles.
Key Responsibilities
- Develop highly scalable and efficient frameworks for different aspects of DL model life cycle
- Design and deploy large scale systems for supervised and unsupervised model training paradigms.
- Understand and address the shortcoming of existing processes for MLOPs and contribute to multiple production data and ML pipelines.
- Collaborate with infrastructure teams to develop tooling and interfaces to support rapid benchmarking, analysis and optimization of our models, hyper-parameter tuning, and data visualization.
Preferred Qualifications
- MS. or Ph.D. in Electrical, Mechanical, or Aerospace Engineering, Computer Science, or a related field.
- 3+ years of experience in training and deploying deep learning models for Computer Vision problems like object detection and segmentation.
- Strong Python programming background and working knowledge of at least one modern DL frameworks amongst PyTorch, Tensorflow, Caffe2 or MXNet.
- Experience with containerization and job management with Kubernetes
- Research and development experience in one or more of the following areas:
- Large scale distributed training pipelines.
- Automating complex MLOps pipelines at regular release cadences.
- Ability to work in a fast-paced development environment
- Be a good team player with great communication skills
- Passionately motivated to take ideas from R&D phase to a productization.
Bonus Skills
- Experience developing and deploying ML pipelines using GCP, Azure ML or AWS SageMaker.
- Prior experience in following industry standard MLOps and DevOps principles in an automotive / ADAS setting.
- Prior experience using convenience DL frameworks like PyTorch lightening or dl-catalyst.
- Familiarity with large scale AutoML pipelines for Neural Network Architectural Search and other training time model compression strategies.
Company Info.
Rivian Automotive, Inc.
Rivian Automotive, Inc., is an American electric vehicle manufacturer and automotive technology company founded in 2009. Rivian is building an electric sport utility vehicle and pickup truck on a skateboard platform that can support future vehicles or be adopted by other companies.
Get Similar Jobs In Your Inbox
Rivian Automotive, Inc. is currently hiring Deep Learning Engineer Jobs in Palo Alto, CA, USA with average base salary of $120,000 - $190,000 / Year.