job details

Back to jobs search
Back to jobs search

Principal Engineer, Frameworks and APIs

GoogleSunnyvale, CA, USADirector+

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.

Information collected and processed as part of your Google Careers profile, and any job applications you choose to submit is subject to Google's Applicant and Candidate Privacy Policy.

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google's EEO Policy, Know your rights: workplace discrimination is illegal, Belonging at Google, and How we hire.

If you have a need that requires accommodation, please let us know by completing our Accommodations for Applicants form.

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

To all recruitment agencies: Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.

Google apps
Main menu