onsite
Staff Machine Learning Engineer, Perception - Rivian
ML Engineer
Lead the design and deployment of advanced perception systems for autonomous vehicles, leveraging deep learning, computer vision, and real‑time inference on embedded platforms.
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 edge hardware.
- Collaborate with sensor integration teams to fuse LiDAR, radar, and camera data, improving robustness in diverse environments.
- Lead performance profiling, model compression, and deployment strategies across C++/Python codebases.
- Mentor junior engineers, conduct code reviews, and drive best practices in ML engineering.
Requirements
- 10+ years of experience in machine learning and computer vision, with a strong background in autonomous vehicle perception.
- Proficiency in Python, PyTorch/TensorFlow, and C++ for high‑performance inference.
- Hands‑on experience with ROS, sensor fusion, and real‑time embedded systems.
- Deep understanding of deep learning architectures (CNNs, Transformers) and training pipelines.
- Excellent communication skills and a track record of leading technical teams.
Skills
machine learningcomputer visiondeep learningpythonpytorchcros