Software Engineer, ML Compilers, Silicon

Google organizes the world's information and makes it universally accessible and useful, combining AI, Software, and Hardware to create helpful experiences.
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
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, Applied AI, Google Cloud

Software Developer III position at Google Cloud focusing on Applied AI and Food AI platform development, requiring Python expertise and ML/NLP experience.

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, AI/ML, Google Cloud AI

Software Developer III position at Google Cloud AI, focusing on developing AI/ML solutions and infrastructure with opportunities to work on cutting-edge technologies that impact billions of users.

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.

Description For Software Engineer, ML Compilers, Silicon

Google is seeking a Software Engineer specializing in ML Compilers to join their EdgeTPU compiler team. This role combines cutting-edge machine learning with compiler optimization, focusing on developing next-generation technologies that will impact billions of users. As part of Google's mission to organize world's information and make it universally accessible, you'll work on critical projects involving compiler development for accelerator-based architectures.

The position offers an opportunity to work at the intersection of hardware and software, optimizing ML workloads for EdgeTPU architecture. You'll be responsible for analyzing and improving compiler quality, developing parallelization algorithms, and collaborating with hardware architects to design future accelerators. This role requires strong expertise in compiler development, machine learning, and system optimization.

The ideal candidate will bring experience in compiler optimizations, particularly for accelerators, and a strong background in ML model optimization for inference. You'll be part of a team that combines the best of Google AI, Software, and Hardware to create radically helpful experiences. This is an excellent opportunity for someone passionate about pushing the boundaries of ML compiler technology and hardware/software co-design.

Working at Google, you'll have the chance to switch teams and projects as you grow, ensuring continuous learning and development. The role offers exposure to various aspects of technology, from distributed computing to system design, and the opportunity to work with cutting-edge ML hardware acceleration technology. Join Google's mission to make computing faster, seamless, and more powerful while making people's lives better through technology.

Last updated 3 months ago

Responsibilities For Software Engineer, ML Compilers, Silicon

  • Work as part of the EdgeTPU compiler team, including analyzing and improving the compiler quality and performance on optimization decisions, correctness, and compilation time
  • Develop parallelization and scheduling algorithms to optimize compute and data movement costs to execute ML workloads on the EdgeTPU
  • 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, ML Compilers, Silicon

  • Bachelor's degree in Computer Science, Electrical Engineering, similar technical field, or equivalent practical experience
  • 2 years of experience with software development in one or more programming languages (e.g., C, C++), or 1 year of experience with an advanced degree
  • 2 years of experience with data structures or algorithms
  • Master's degree or PhD in Computer Science or related technical field (preferred)
  • Compiler development experience in the context of accelerator-based architectures (preferred)
  • Experience with compilers and compilers for accelerators (preferred)
  • Experience in optimizing ML models for inference (preferred)

Interested in this job?