Plus.ai

Senior/Staff Machine Learning Engineer (Reinforcement Learning), Motion Planning

Plus.ai • $130K — $220K *
Santa Clara, CA 95051In-Person
Transportation
Less than 5 years of experience
Today
Be an Early Applicant
Job Overview by Ladders

Qualifications

  • MS or PhD in Computer/Software Engineering, Robotics, Machine Learning or related field
  • Experience developing production code in C/C++ and Python
  • Experience with machine learning frameworks such as PyTorch
  • Experience with reinforcement learning
  • Strong oral and written communication skills.

Responsibilities

  • Design, develop, and deploy learned behavior planning models using reinforcement learning and behavior cloning
  • Own the end-to-end Reinforcement Learning pipeline including environment construction and modeling
  • Contribute to Reinforcement Learning Infrastructure for large-scale distributed training
  • Track and analyze model training performance, improving generalization across driving scenarios
  • Collaborate with simulation and planning teams to integrate models into driving pipelines

Benefits

  • Work in a future-oriented, innovative field
  • Opportunities for personal and professional development
  • Catered free lunch and unlimited snacks
  • Highly competitive salary and benefits package including 401(k) plan
Full Job Description
Responsibilities

  • Design, develop, and deploy learned behavior planning models using approaches such as reinforcement learning, behavior cloning, and imitation learning
  • Own end-to-end Reinforcement Learning pipeline from environment construction, training, and modeling.
  • Contribute to Reinforcement Learning Infrastructure to facilitate large-scale distributed training
  • Track model training, analyze model performance, identify failure modes to improve generalization across driving scenarios
  • Collaborate with simulation and planning teams to deploy learned behavior planning models to driving pipelines


Required skills

  • MS or PhD in Computer/Software Engineering, Robotics, Machine Learning or related field.
  • Experience developing production code in C/C++ and Python
  • Experience with machine learning frameworks such as PyTorch
  • Experience with reinforcement learning
  • Strong oral and written communication skills.


Preferred skills

  • Experience with high-throughput batched driving simulators such as GPUDrive and PufferDrive
  • Experience with self-play reinforcement learning
  • Experience with sim-to-real in the context of robotics / autonomous driving
  • Prior experience in learned behavior planning / prediction for autonomous vehicles


$130,000 - $220,000 a year

Your opportunities joining PlusAI

Work, learn and grow in a highly future-oriented, innovative and dynamic field.

Wide range of opportunities for personal and professional development.

Catered free lunch, unlimited snacks and beverages.

Highly competitive salary and benefits package, including 401(k) plan.

About Plus.ai

Land O'Lakes, Inc. is an American member-owned agricultural cooperative based in the Minneapolis-St. Paul suburb of Arden Hills, Minnesota, United States, focusing on the dairy industry. The cooperative has 1,959 direct producer-members, 751 member-cooperatives, and about 9,000 employees who process and distribute products for about 300,000 agricultural producers, handling 12 billion pounds of milk annually. It is ranked third on the National Cooperative Bank Co-op 100 list of mutuals and cooperatives. The co-op is one of the largest producers of butter and cheese in the United States through its dairy foods business; serves producers, animal owners and their families through more than 4,700 local cooperatives, independent dealers and other large retailers through its Purina Animal Nutrition business; and delivers seed, crop protection products, agricultural services and agronomic insights to 1,300 locally owned and operated cooperative and independent agricultural retailers and their grower customers through its WinField United business.
Learn more about Plus.ai
Industry
Manufacturing & Automotive
* Ladders Estimates

Similar Jobs

Ad banner background

Get Ready For Your
Next Interview

More Jobs at Plus.ai

More Transportation Jobs

Find similar Senior/Staff Machine Learning Engineer (Reinforcement Learning), Motion Planning jobs:

NationwideSanta Clara, CA