rivian·Machine Learning Engineer·Technical Phone Screen
May 2026
A 45-minute coding round for an MLE role at Rivian, entirely pandas-based with a small pre-shipped codebase. Three parts, progressively harder, no LeetCode nonsense which was a relief.
- Given a small data-processing pipeline with a bug in it, identify and fix the issue so the pipeline runs correctly.
- Implement a helper function that aggregates and transforms a time series, covering things like resampling, rolling window features, and lag generation using pandas.
- Using the prepared time-series data from the previous step, implement a prediction function that produces forecasts.
“Part one felt like a warmup but I almost overthought it.”