Principal Engineer, Frameworks and APIs

Sunnyvale, California


Employer: Google
Industry: Business Strategy
Salary: Competitive
Job type: Full-Time

Minimum qualifications:

  • Bachelor's degree in Computer Science, a similar technical field, or equivalent practical experience
  • 15 years of experience as a software engineer or 13 years with an advanced degree
  • Experience in applying machine learning (ML) to solve problems
  • Experience with technical innovation in Frameworks, Libraries, Tools, APIs, or related fields


Preferred qualifications:

  • Experience in ML technologies
  • Experience in Open Source development
  • Technical expertise in systems and software with the leadership skills to influence technical leaders across the company
  • Understanding of the needs of advanced ML developers
  • Excellent API and systems design skills


About the job

Google's Core Machine Learning (ML) team aims to build AI platforms, services, and tools behind Google's AI research and AI-empowered products. We are owners of the Google AI framework (e.g., TensorFlow, JAX), AI performance and efficiency, AI training and inference platforms, AI compilers (e.g. XLA), AI software/hardware co-design (e.g. TPU), and AI developer experience (e.g., Model Hub, Eval Hub). Further, we are the driver to launch LMs (e.g. large models such as Gemini) from research to production and enable all Google products to adopt LMs to improve end user experiences. The Core ML team is responsible for planning and optimizing ML capacity for all Google products and driving performance and efficiency work to ensure we maximize return on investment for all our AI investments. We look across Google's AI products and AI research to build central solutions, break down technical barriers, and strengthen existing systems.

As the Principal Engineer, Frameworks and APIs, you will lead the design and development of the Core ML frameworks and APIs. You will be a Technical Lead helping design and develop the core ML API stack and ecosystem, extending Google's leadership position in ML. You will work closely with TensorFlow and JAX leads on designing and building sustainable APIs for ML developers at Alphabet and throughout the world.

Google Cloud accelerates every organization's ability to digitally transform its business and industry. We deliver enterprise-grade solutions that leverage Google's cutting-edge technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems.

The US base salary range for this full-time position is $278,000-$399,000 bonus equity benefits. Our salary ranges are determined by role, level, and location. The range displayed on each job posting reflects the minimum and maximum target salaries for the position across all US locations. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training. Your recruiter can share more about the specific salary range for your preferred location during the hiring process.

Please note that the compensation details listed in US role postings reflect the base salary only, and do not include bonus, equity, or benefits. Learn more about benefits at Google .

Responsibilities

  • Provide leadership in designing and implementing user-friendly, high-performance low-level APIs for ML production systems in C and Python.
  • Develop a structure of other Technical Leads in the area, both by defining technical goals and orienting teams around technical decisions they can make, and by providing development support to Engineers in the area.
  • Escalate when needed to Directors or Vice Presidents about the lack of available Technical Leads, lack of resources, or challenges in development for the team.

Created: 2024-06-05
Reference: 77377536761176774
Country: United States
State: California
City: Sunnyvale
ZIP: 95002