Principal Engineer, Foundations API and Runtime, Core ML

Key Skills

Java Programming, Python Programming, SQL, Infrastructure as code, Dash framework

Job Description

Minimum qualifications:

• Bachelor's degree in computer science, mathematics, other relevant engineering field, or equivalent practical experience
• 15 years of experience as a Software Engineering leader in Infrastructure
• Experience motivating others at all levels by creating a vision
• Experience growing and scaling teams

Preferred qualifications:

• Experience building ML infrastructure or products with heavy ML usage
• Experience building software runtimes or intermediate representations of programs in an ML environment
• Track record of ideation and innovation of technology at scale and passion for development and the use of cross-platform shared code
• Understanding of computing, data centers, enterprise software, and hardware infrastructure with technical knowledge to be credible with customers and engineers
• Ability to balance detailed, technical guidance with big picture strategy, enabling teams to deliver products that are effective and also creating ways to manage data

About the job

As a Principal Engineer for Foundations API and Runtime team at Core ML, you will be responsible for the technical design and vision for enabling a framework agnostic foundational infrastructure stack, from compilers through to devices. The team supports a wide range of deployment environments, from user facing ML serving services through to a variety of ML frameworks across devices, such as large, supercomputers to mobile edge. You will be deeply involved in setting up long-term strategy for ML framework development at Google, and developing new foundations for emerging ML requirements, such as models, sparsity and quantization across server, mobile and edge.

The Core team builds the technical foundation behind Google’s flagship products. We are owners and advocates for the underlying design elements, developer platforms, product components, and infrastructure at Google. These are the essential building blocks for excellent, safe, and coherent experiences for our users and drive the pace of innovation for every developer. We look across Google’s products to build central solutions, break down technical barriers and strengthen existing systems. As the Core team, we have a mandate and a unique opportunity to impact important technical decisions across the company.

Responsibilities

• Develop a strategic vision for the low-level foundations of ML at Google. Lead the design of APIs to support future ML development.
• Work with partners from ML frameworks (TensorFlow, JAX), Product Areas (Ads, Search, YouTube, Android), and hardware (CPU, GPU, TPU, Edge TPU) to develop APIs which support the full lifecycle of ML at Google. Influence partners and stakeholders within and across the organization to build joint roadmaps and drive adoption of APIs.
• Influence external ML framework developer roadmaps via direct engagements as well as Operational Support Systems (OSS) strategy. 
• Lead the technical design across ML and infrastructure engineers as needed to build the next generation of APIs.
• Lead, design, and develop the Foundations API strategy in alignment with Core ML strategy.