Staff Machine Learning Engineer

What you'll do

• Research reinforcement-learning algorithms and design efficient architectures for real-time vehicle planning.
• Build offline, data-driven infrastructure and use fleet and field-testing data to improve planning models.
• Evaluate solutions through experiments and testing, and collaborate with perception and systems teams to ship production ML models.

What they ask for

• Education in Engineering, Robotics, Computer Science, or a related field focused on Reinforcement Learning or Artificial Intelligence, or equivalent experience.
• Strong applied reinforcement learning experience, including model architecture design, training, data mining, and analytics.
• 3-5 + years of experience with machine learning frameworks such as PyTorch, Tensorflow.
• Strong Python programming and software design skills; solid understanding of data structures, algorithms, code optimization, and large-scale data processing.

Nice to have

• MS or PhD-level education in a relevant field, or equivalent experience; autonomous-driving or robotics production experience.
• Hands-on experience developing RL-based planning engines; experience with ONNX and TensorRT.

Pay

$179,400 - $303,600 USD base salary

Tools

PyTorch, Tensorflow, Python, ONNX, TensorRT

AI focus

Reinforcement learning for autonomous-driving planning, including model development, real-time deployment, and data-driven adaptation.