ML Engineer, Behavior Planning

Required

• BS, MS, or PhD in Robotics, Machine Learning, Computer Science, or a related quantitative field, or equivalent practical experience.
• 2+ years of experience developing state-of-the-art ML models, either in a research or production setting.
• Hands-on experience working on Imitation Learning or Reinforcement Learning applied to physical or simulated agents.
• Experience training large models on massive datasets using distributed computing.
• Fluency in Python, with a focus on writing high-performance, maintainable code.
• Deep experience with PyTorch (preferred) or another modern ML framework, and a mastery of modern ML architectures including Transformers and Diffusion Models.
• Desirable qualifications
• A track record of publications in top-tier ML conferences (NeurIPS, ICML, CoRL, CVPR, AAAI).
• Experience deploying complex ML systems in production environments.
• Experience in developing generative models or neural simulators for synthetic data generation.
• Fluency with C++, with a focus on writing high-performance, safe, and maintainable code.
• The base salary range for this position is $139,000 - $223,000 per Year. Aurora’s pay ranges are determined by role, level, and location. Within the range, the successful candidate’s starting base pay will be determined based on factors including job-related skills, experience, qualifications, relevant education or training, and market conditions. These ranges may be modified in the future. The successful candidate will also be eligible for an annual bonus, equity compensation, and benefits.