6 months ago
Base Salary
$168k - $247k/yr
Responsibilities
- Formulate complex driving tasks as reinforcement learning problems with reward functions and state/action representations
- Design and train model-based deep reinforcement learning agents using GPU-accelerated simulation at massive scale
- Improve the reinforcement learning simulator
- Build and maintain distributed training infrastructure in JAX across large compute clusters
- Build agentic optimization systems that automate code improvements, experiments, metric analysis, and policy iteration
- Own the lifecycle of learned policies from problem formulation and reward design through distributed training and on-vehicle inference
- Define how learned components integrate with the broader autonomy stack to produce robust, shippable behavior
Requirements
- BS, MS, or PhD in computer science, electrical engineering, robotics, or a related field
- Strong foundation in reinforcement learning and deep learning
- Hands-on experience training reinforcement learning agents at scale
- Proficiency in JAX or a similar functional machine learning framework
- Comfort with JIT compilation, vectorized environments, and GPU-accelerated simulation
- Strong knowledge of policy gradients, value functions, exploration-exploitation, model-based reinforcement learning, reward shaping, and sim-to-real transfer
- Experience building experiment pipelines, analyzing training runs, and using metrics to guide architectural decisions
- Preferred experience in robotics, autonomous driving, or other real-time decision-making domains
- Preferred publications at venues such as NeurIPS, ICML, ICLR, CoRL, RSS, or ICRA
- Preferred experience building or using GPU-accelerated simulators for reinforcement learning
- Preferred track record shipping learned components in production robotics or autonomous vehicle stacks
Benefits
- 401(k) plan with employer matching
- 16 weeks of paid parental leave
- Wellness benefits and expense reimbursement
- Commuter benefits match
- Paid time off and paid sick leave
- Medical, dental, and vision benefits
- 11 paid holidays
- Disability and basic life insurance
- Family-forming assistance
- Mental health program
- Equity grant opportunities
- Salaried roles include flexible paid time off and 80 hours of paid sick time per year
Tech Stack
Categories
ML EngineeringRobotics
About DoorDash
DoorDash builds a marketplace and logistics platform that connects consumers with local restaurants and retailers for on-demand delivery and pickup, powered by a network of independent Dashers. It earns through delivery fees, merchant commissions, advertising, DashPass subscriptions, and white-label fulfillment via DoorDash Drive. Founded in 2013 and headquartered in San Francisco, the public company operates across the U.S. and selected international markets, and also researches automation through DoorDash Labs.