Remote Jobs RockRemote Jobs Rock

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

📅 Aug 27
Reinforcement LearningBehavior CloningImitation LearningC/C++
Apply Now

📜 Description

  • Design, develop, and deploy learned behavior-planning models using reinforcement learning, behavior cloning, and imitation learning.
  • Own the reinforcement learning pipeline, from environment construction through training and modeling.
  • Contribute to infrastructure for large-scale distributed reinforcement learning training.
  • Track training and model performance, identify failure modes, and improve generalization across driving scenarios.
  • Collaborate with simulation and planning teams to integrate learned models into driving pipelines.

🛠️ Requirements

  • 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.
  • 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

✨ Benefits

  • Competitive salary and benefits package.
  • 401(k) plan.
  • Catered free lunch.
  • Unlimited snacks and beverages.
  • Opportunities for personal and professional development.
Full job description
PlusAI is a Physical AI company pioneering AI-based virtual driver software for factory-built autonomous trucks. Headquartered in Silicon Valley with operations in the United States and Europe, Plus was named by Fast Company as one of the World’s Most Innovative Companies. Partners including TRATON GROUP’s Scania, MAN, and International brands, Hyundai Motor Company, Iveco Group, Bosch, and DSV are working with Plus to accelerate the deployment of next-generation autonomous trucks. If you’re ready to make a huge impact and drive the future of autonomy, Plus is looking for talented individuals to join its fast-growing teams.
PlusAI is a Physical AI company pioneering AI-based virtual driver software for factory-built autonomous trucks. Headquartered in Silicon Valley with operations in the United States and Europe, Plus was named by Fast Company as one of the World’s Most Innovative Companies. Partners including TRATON GROUP’s Scania, MAN, and International brands, Hyundai Motor Company, Iveco Group, Bosch, and DSV are working with Plus to accelerate the deployment of next-generation autonomous trucks. If you’re ready to make a huge impact and drive the future of autonomy, Plus is looking for talented individuals to join its fast-growing teams. 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.

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
  • Wayve

    Senior Machine Learning Engineer, Driving Product

    🕒 7 days ago
    Wayve👥 501 - 1000 employees🏢 Computer Software

    Develop and ship autonomous-driving ML models for L2–L4 vehicles, owning data curation, training, evaluation, and on-road iteration to improve driving behaviours and real-world performance.

    PythonDeep LearningMachine Learning FrameworksAutonomous Driving
    GoDaddy

    Senior Software Development Engineer

    🕒 2 days ago
    GoDaddy👥 5001 - 10,000 employees🏢 Internet

    Build and operate production machine-learning pipelines and scalable ML systems for fraud mitigation, customer insights, and AI-driven commerce experiences, from architecture and deployment through monitoring and iteration.

    PythonAWS Sagemaker StudioSagemaker PipelinesTecton
    Scandit

    Senior Machine Learning Engineer - Video Understanding

    🕒 4 days ago
    Scandit👥 201 - 500 employees🏢 Computer Software

    Lead technical discovery and development of video-based computer vision systems for retail, building end-to-end ML pipelines and action-recognition models while validating performance and production constraints.

    Computer VisionMachine LearningVideo UnderstandingAction Recognition

    Trusted by Remote Workers