Senior Machine Learning Infrastructure Engineer for Autonomous Driving Systems

6 days ago


Palo Alto, California, United States Rivian Full time
About Rivian

Rivian is on a mission to keep the world adventurous and environmentally sustainable. The company specializes in designing and manufacturing electric adventure vehicles, showcasing its commitment to innovation and customer satisfaction.

Compensation and Benefits

The estimated annual salary for this position in California ranges from $206,500 to $258,100, based on experience and other factors.

Job Summary

In this role, you will collaborate with cross-functional teams to develop infrastructure for state-of-the-art machine learning models used in ADAS systems. This involves working closely with software, hardware, system, and SoC groups to create efficient building blocks for advanced machine learning models.

Responsibilities:
  • Map various versions of ADAS perception models (development, production, prototype) to SoC, transforming graphs and operators while ensuring efficient performance on SoC.
  • Create or improve existing QAT infrastructure.
  • Quantize models to different data precisions (4b, 8b, 16b) while preserving originally FP32 accuracy.
  • Compress models using structured sparsity schemes.
Qualifications:
  • Bachelor's degree in Electrical Engineering, Computer Science, or a relevant field.
  • Expertise or prior work experience in one of the following areas: Quantization of Neural Networks, Sparsity of Neural Networks, Training of Neural Networks.
  • Experience with ADAS-specific workloads dealing with Nuscenes, Waymo datasets is a plus.
  • Demonstrate proficiency in C++, Python, Numpy, PyTorch framework.
  • Excellent debugging and analytical skills, written and verbal communication skills, and a strong team player.

Rivian provides robust medical/Rx, dental, and vision insurance packages for full-time employees and their dependents. Coverage begins on the first day of employment, and most premiums are covered by Rivian.

Rivian is an equal opportunity employer, committed to creating an inclusive environment that values diversity and fosters growth.



  • Palo Alto, California, United States Match Group Full time

    About the OpportunityWe are seeking a highly skilled Sr. Software Engineer, Machine Learning Infrastructure to join our team at Match Group. This role will be responsible for building and maintaining scalable infrastructure to support machine learning engineers across various business units.The TeamYou will work closely with cross-functional teams of...


  • Palo Alto, California, United States Snap Full time

    Company OverviewSnap Inc is a technology company that believes the camera presents the greatest opportunity to improve the way people live and communicate. The Company's three core products are Snapchat, an app that enhances your relationships with friends, family, and the world; Lens Studio, an augmented reality platform that powers AR across services; and...


  • Palo Alto, California, United States Pinterest Full time

    Pinterest is seeking a highly skilled Senior Machine Learning Engineer to join our Monetization Engineering team. As a key member of this team, you will be responsible for developing and executing a vision for the evolution of the machine learning technology stack within Ads.This is an exciting opportunity to work on large-scale recommendation systems,...


  • Palo Alto, California, United States HP Development Company, L.P. Full time

    This key leadership role oversees a team of machine learning engineers and data scientists, guiding model development and aligning with business goals.The Senior Machine Learning Architect designs robust models to support business decisions and strategies. The role triages new AI/ML use cases and formulates a plan for delivery. The role communicates...


  • Palo Alto, California, United States Tesla Full time

    As a key member of the Tesla Bot team, you will play a vital role in developing and implementing cutting-edge machine learning infrastructure. This position offers an exceptional opportunity to design, build, and deploy scalable solutions for neural network architecture, data visualization, model export, and deployment.ResponsibilitiesDesign and improve...


  • Palo Alto, California, United States Rivian Full time

    Rivian, a pioneer in electric adventure vehicles, seeks an exceptional Perception Data Engineer to join their Autonomy team. This role offers the opportunity to design, implement, and maintain large-scale data infrastructure that drives machine learning development for safety-critical self-driving features.Job DescriptionIn this position, you will work...


  • Palo Alto, California, United States Amazon Full time

    About the RoleWe are seeking an experienced Data Scientist to join our Search Science Data Infrastructure team at Amazon. As a Data Scientist, you will play a key role in designing, developing, and deploying big data and machine learning services that power our search engine.Key ResponsibilitiesYou will lead the development of services and infrastructure at...


  • Palo Alto, California, United States OPPO US Research Center Full time

    Job OverviewThe OPPO US Research Center seeks a highly skilled Machine Learning Engineer to drive innovation and advancement in our research and development team. This role offers an exciting opportunity to contribute to the refinement of our recommender system, search algorithms, and ad targeting algorithms.


  • Palo Alto, California, United States Robotics Prcocess Automation, LLC Full time

    Job OverviewRobotics Process Automation, LLC is seeking an experienced Senior Machine Learning Research Scientist to join our team. The ideal candidate will have a Ph.D. degree in Artificial Intelligence, Machine Learning, Robotics, or a related field with a strong publication record.Key ResponsibilitiesThe successful candidate will be responsible...


  • Palo Alto, California, United States OPPO US Research Center Full time

    As a key member of the OPPO US Research Center's technology innovation leadership team, we are seeking an exceptional Machine Learning Engineer to join our Research and Development team. This role is crucial in refining our recommender system, search algorithms, and ad targeting algorithms.About UsThe OPPO US Research Center is a fast-growing innovation hub...


  • Palo Alto, California, United States Pinterest Full time

    About the RolePinterest is a platform where millions of people come to find new ideas every day. As a Staff Machine Learning Engineer on our Content Understanding team, you will play a crucial role in developing horizontal knowledge graph and content understanding signals.You will be responsible for building content signals that power personalized product...


  • Palo Alto, California, United States Pinterest Full time

    About us:Pinterest is a platform where millions of people across the world come to find new ideas every day. Our mission is to help them find their inspiration and create a life they love.In your role, you will be challenged to take on work that upholds this mission and pushes Pinterest forward. You will grow as a person and leader in your field, all while...


  • Palo Alto, California, United States Acceler8 Talent Full time

    Acceler8 Talent Seeks Founding Machine Learning Research EngineerWe are seeking enthusiastic individuals to join our pioneering team as Founding ML Research Engineers. This role offers both junior and senior opportunities, allowing individuals at different stages of their career to contribute to groundbreaking projects.About the Role:Develop and evaluate...


  • Palo Alto, California, United States Amazon Full time

    Company OverviewWe are an innovative team at Amazon, dedicated to improving the shopping experience using conversational large language models. Our team is passionate about tech, innovation, and customer satisfaction, with a strong background in machine learning.About the RoleWe are seeking experienced engineers to work with our scientists and engineers to...


  • Palo Alto, California, United States hp Full time

    Are you a visionary seeking to revolutionize the world of artificial intelligence? We're looking for a talented Machine Learning Engineering Intern to join our dynamic team at HP.About Our OpportunityWe are committed to fostering a diverse and inclusive workplace that attracts exceptional talent and supports employees in their success. As an intern, you'll...


  • Palo Alto, California, United States EarnIn Full time

    At EarnIn, we're pioneers in earned wage access, dedicated to delivering real-time financial flexibility for individuals living paycheck to paycheck. Our products empower community members to access their earnings as they earn them, without mandatory fees, interest rates, or credit checks.We have a world-class leadership team and top-notch funding partners...


  • Palo Alto, California, United States Pinterest Full time

    About the RoleWe're seeking a seasoned Engineering Manager to lead our machine learning team in developing and executing a vision for the evolution of the technology stack for ads engagement models. The ideal candidate will have experience in building production machine learning systems at scale, data mining, search, recommendations, and/or natural language...


  • Palo Alto, California, United States Qualified Health Full time

    Job Overview:Achieve excellence as a Senior Data Science Leader in our innovative healthcare team, focusing on AI-driven solutions that transform patient care and outcomes.About the Role:We are seeking a highly skilled and experienced Senior Data Science Leader to drive technical strategy, lead cross-functional teams, and deliver impactful machine learning...


  • Palo Alto, California, United States Gauss Labs Full time

    Job DescriptionGauss Labs is seeking an experienced Lead Software Architect for Machine Learning to lead the development and launch of core software products.About the RoleThis is a challenging and rewarding opportunity to work alongside talented Software Engineers and AI Scientists. As a Lead Software Architect, you will have the opportunity to deepen your...


  • Palo Alto, California, United States OPPO US Research Center Full time

    Unlocking Innovation in AI TechnologyWe are seeking a highly skilled Senior Machine Learning Engineer to join our Research and Development team at the OPPO US Research Center. This exciting role offers a unique opportunity to refine our recommender system, search algorithms, and ad targeting algorithms, pushing the boundaries of machine learning technology.