Taro Logo

Software Engineer, Machine Learning, Edge TPU

Google develops next-generation technologies for information organization and accessibility, focusing on AI, software, and hardware integration.
$136,000 - $200,000
Machine Learning
Mid-Level Software Engineer
2+ years of experience
AI
This job posting may no longer be active. You may be interested in these related jobs instead:
Software Developer III, AI/ML GenAI, Applied AI

Software Developer III position at Google focusing on AI/ML and GenAI implementation, requiring 2 years of software development experience and expertise in machine learning infrastructure.

Software Developer III, AI/ML Natural Language Processing, Google Workspace

Software Developer III position at Google focusing on AI/ML Natural Language Processing for Google Workspace, combining software engineering with machine learning expertise.

Software Developer III, Google Cloud AI

Software Developer III position at Google Cloud AI, focusing on developing AI solutions for financial services industry, requiring 2+ years of software development experience.

Cloud Engineer II, AI/ML, Professional Services (English, Spanish)

Cloud Engineer II position at Google Cloud focusing on AI/ML solutions, requiring 6 years of ML experience and fluency in English and Spanish.

App Development Customer Engineer, Global Partnerships Sellside

App Development Customer Engineer role at Google focusing on AI/ML solutions for EMEA partners, requiring 6 years of experience and deep technical expertise.

Description For Software Engineer, Machine Learning, Edge TPU

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day.

You will work as part of the EdgeTPU Applied Machine Learning team. We are leading the efforts of defining, developing and training edge optimized models for Generative AI, computer vision, natural language, and speech use cases. You will design, build, and maintain model optimization tools and infrastructure modules needed for automating optimization and training of neural networks and architecture design space exploration. Additionally, you will closely collaborate with ML model developers, researchers, and EdgeTPU hardware/software teams to accelerate the transition from research ideas to user experiences running on the EdgeTPU.

Google's mission is to organize the world's information and make it universally accessible and useful. Our team combines the best of Google AI, Software, and Hardware to create radically helpful experiences. We research, design, and develop new technologies and hardware to make computing faster, seamless, and more powerful. We aim to make people's lives better through technology.

Last updated 7 months ago

Responsibilities For Software Engineer, Machine Learning, Edge TPU

  • Design, build, and maintain model optimization tools and infrastructure modules needed for automating optimization and training of neural networks and architecture design space exploration.
  • Provide cross-team support for compiler, architecture exploration, and neural network design.
  • Write modular and efficient machine learning training pipelines and assist in building profiling and visualization tools.
  • Use existing and newly developed tools to drive ML optimization infrastructure.
  • Work with EdgeTPU architects to design future accelerators, the hardware/software interface, and co-optimizations of the next generation EdgeTPU architectures.

Requirements For Software Engineer, Machine Learning, Edge TPU

Python
  • Bachelor's degree or equivalent practical experience.
  • 2 years of experience with software development in one or more programming languages (e.g., C++, Python), and with data structures or algorithms.
  • 2 years of experience with machine learning algorithms and tools (e.g. TensorFlow), artificial intelligence, deep learning, or natural language processing.
  • Experience with Large Language Models, NLP, or Generative AI.

Benefits For Software Engineer, Machine Learning, Edge TPU

  • bonus
  • equity
  • benefits

Interested in this job?