Principal Software Engineer, AI Platform (ADAS) - Warren, MI AND Mountain View, CA

APPLY NOW

Description
We are seeking a highly skilled AI/ML Platform Engineer to join our team. In this role, you will be responsible for designing, developing, and maintaining our AI/ML platform infrastructure. You will work closely with data scientists, machine learning engineers, and software developers to enhance our platform's capabilities for training complex models on large datasets efficiently. Your contributions will directly impact our ability to scale and deploy cutting-edge AI solutions.

• Foundational AI/ML platform functionality development such as designing and developing scalable infrastructure to support training of machine learning models on large datasets.
• Implementing mechanisms to monitor and manage resource utilization across multi-GPU on-prem and cloud environments.
• Enhance machine learning platform abstractions and APIs to improve usability and productivity for data scientists and engineers.
• Develop and maintain tools for model versioning, experiment tracking, and reproducibility.
• Collaborate with cross-functional teams to integrate new features and technologies into the platform.

• Bachelors or higher degree in Computer Science or equivalent major
• 8+ years professional software engineering experience
• 3+ years specialized experience in AI/ML infrastructure, e.g., enabling distributed training for scaling large ML models
• Strong programming skills in Python, with proficiency in frameworks such as TensorFlow, PyTorch, or similar
• Experience with distributed computing, GPU computing, and cloud environments (AWS, GCP, Azure).
• Willingness to travel to Mountain View as needed
• Comfortable working in highly ambiguous and dynamic environments