onsite
Senior/Staff Machine Learning Engineer - Perception - Rivian
ML Engineer
Lead the design and deployment of perception systems for autonomous vehicles, leveraging advanced computer vision, deep learning, and real‑time inference on edge hardware.
About the role
Key Responsibilities
- Architect and implement end‑to‑end perception pipelines for autonomous driving, including object detection, tracking, and semantic segmentation.
- Develop and optimize deep learning models using PyTorch/TensorFlow, ensuring high accuracy and low latency on embedded platforms.
- Collaborate with sensor integration teams to fuse LiDAR, radar, and camera data, improving robustness in diverse environments.
- Lead code reviews, mentor junior engineers, and drive best practices in software engineering and model validation.
- Conduct research on novel perception algorithms, publish findings, and translate research into production‑ready solutions.
Requirements
- 10+ years of experience in machine learning and computer vision, with a strong portfolio of production‑grade perception systems.
- Proficiency in Python, C++, and deep learning frameworks (PyTorch/TensorFlow).
- Hands‑on experience with ROS, real‑time inference, and deployment on automotive hardware.
- Excellent problem‑solving skills, ability to work in a fast‑paced, cross‑functional team.
- Strong communication skills and a passion for advancing autonomous vehicle technology.
Skills
machine learningcomputer visiondeep learningpythonpytorchcros