Principal - ML Architect

Job Overview

As a Principal Architect, you will lead the design and evolution of intelligent, scalable systems for Multi-Agent Path Finding (MAPF). You’ll define architectural direction, integrate machine learning and reinforcement learning, and build the core capabilities required to achieve safe, efficient, and autonomous decision-making at scale.

Key Responsibilities

• Architect distributed systems to support real-time path planning of multi-agent using Machine Learning.
• Lead design and implementation of ML model pipelines, including data ingestion, training, validation, deployment, and monitoring.
• Own production deployment of ML/RL models using MLOps tools such as Vertex AI or similar platforms.
• Integrate ML pipelines with robotic orchestration engines to support continuous learning and adaptation.
• Collaborate closely with software, robotics, product, and operations teams to align system goals with real-world fulfillment challenges.
  
  

Required Qualifications

• Education: B.E/M.S in Computer Science, AI/ML, or a related field.
• Experience: 12 years of total experience (7 years of experience in system architecture & AI/ML ; including 2 years in technical leadership roles).
  
  

Technical Expertise

• Strong knowledge of path planning, graph search algorithms, and optimization techniques for multi-agent systems.
• Deep understanding of machine learning, deep learning, and reinforcement learning, with experience using TensorFlow or PyTorch.
• Proven experience in building, deploying, and maintaining ML models in production environments.
• Hands-on experience with MLOps, including CI/CD for models, pipeline orchestration, and model monitoring.
  
  

California Pay Range: $180,000 USD - $200,000 USD