Skip to content

Search jobs inside your AI agent.

Grok Bot

Connect

Pony.ai

Machine Learning Engineer (Reinforcement Learning)

Posted Updated
Discussion

Description

  • Build scalable systems for training and fine-tuning large generative models that produce realistic, informative driving behaviors for evaluation and scenario coverage
  • Implement and iterate on RL-style methods: algorithms, reward / preference objectives, and training setups suited to high-fidelity, insightful behaviors in simulation-aligned workflows (closed-loop evaluation mindset)
  • Ship deep learning solutions (including LLM / VLM where appropriate) that improve human-led triaging, automate high-volume workflows, and support nuanced analysis of self-driving behavior to surface critical anomalies
  • Own production-oriented ML for fleet-scale assessment: training, optimization, monitoring, and iteration of models used to judge performance across large real-world exposure
  • Design and evolve data + evaluation systems inspired by RL from human preferences (RLHF) and related paradigms—turning preference/judgment signals into repeatable, scalable training and evaluation loops
  • Partner broadly with teams such as Prediction, Planning, Research, and platform/engineering leads to land cross-cutting improvements with clear metrics

Benefits

  • Comprehensive medical, dental, and vision insurance
  • Generous PTO and paid holidays
  • Daily meals and bi-weekly happy hours
  • Equity opportunity
  • Retirement plans
  • Frequent tech talks
  • Hands-on experience building and applying ML in production-grade settings, with a strong RL component (policy learning, preference/feedback optimization, or offline/online RL pipelines)
  • Experience with large-scale distributed training and large-scale data processing
  • Ability to lead ambiguous technical work from problem framing through reliable delivery
  • M.S. or Ph.D. in Computer Science, Machine Learning, AI, or a related field—or equivalent practical experience
  • Demonstrated impact via strong publications or a clear history of shipping impactful ML systems end-to-end
  • Depth in deep learning, sequence modeling, and generative models
  • Background in autonomous vehicles, robotics, or complex simulation environments
  • Strong grasp of modern RL and post-training techniques in LLM, dLLM, VLA and video generations
  • Hands-on integration of simulation platforms with ML training and evaluation workflows
  • Python fluency and frameworks such as PyTorch
  • Experience defining and operating metrics for complex, safety-critical AI systems
  • Technical leadership: influencing stakeholders, aligning teams, and raising the bar for evaluation rigor
  • Excellent communication—simple explanations of complex trade-offs

Skills

Apply

See also

ML / AI jobs by country — openings, pay and top skills →