Rivian interview question

ML focused coding question and conceptual