Senior Machine Learning Engineer - ML Agents and Planning
Develops ML models using imitation and reinforcement learning for agent behavior prediction and autonomous vehicle planning. Requires PhD +1 year or MSc +5 years experience, expertise in RL, transformers, and production ML pipelines.
Responsibilities
- Develop deep learning models using imitation learning and reinforcement learning for human-like agent driving plans.
- Create novel techniques to estimate driving plan quality (safety, progress, comfort, realism).
- Contribute to large-scale ML infrastructure to discover solutions and advance the field.
- Develop metrics and tools to analyze errors and improvements.
- Collaborate with Perception, Planning, Simulation, and Validation engineers on autonomous driving.
Qualifications
- PhD in computer science or related field +1 year professional experience (top-tier publications may substitute), or MSc +5 years professional experience.
- Experience in Planning/Prediction using Reinforcement Learning.
- Experience training and deploying transformer-based models.
- Experience with production ML pipelines: dataset creation, training frameworks, metrics.
- Fluency in Python; basic C++ understanding.
Bonus Qualifications
- Top-tier publications (NeurIPS, ICML, CVPR).
Senior Staff Machine Learning Engineer, Communication & Connectivity
Lead ML architecture and implementation for Airbnb's Messaging & Notifications, building recommendation engines, ranking systems, and LLM-powered experiences while mentoring engineers.
Staff Software Engineer
Founding Staff Applied Agent Engineer to architect and lead Traba's agentic platform, building production LLM/agent systems that integrate with customer WMS/TMS/ERP and drive industrial operations. Requires 7+ years engineering experience with 2+ years building production agent systems.
Senior Software Engineer
Founding Senior Applied Agent Engineer building production LLM agent systems that automate supply chain workflows. Requires 5+ years engineering experience with 1+ year shipping LLM/agent features, strong Python/TypeScript skills, and hands-on agent stack experience.
Staff Software Engineer, Cribl AI
Staff-level AI/ML engineer building and productionizing generative AI features across backend and frontend for Cribl's observability platform. Requires 6+ years experience, AI/ML and MLOps background, and TypeScript/JavaScript proficiency.