Software Engineer, ML Compiler

2 Hours ago β€’ 2-5 Years β€’ Research & Development

About the job

Job Description

Google Cloud seeks a Software Engineer for its ML Compiler team. The role involves profiling and analyzing compiled ML model performance, improving code generation within XLA GPU or Triton compilers, and participating in rotations addressing bugs, guiding external contributors, or integrating MLIR. Minimum qualifications include a Bachelor's degree in a related field and 2 years of software development experience with data structures/algorithms, along with 2 years in ML infrastructure (model deployment, evaluation, data processing). Preferred qualifications include 3 years of compiler optimization experience. The engineer will contribute to critical Google Cloud projects, with opportunities for team and project changes. The ideal candidate will be versatile, display leadership, and proactively tackle full-stack challenges.
Must have:
  • Bachelor's degree in CS or related field
  • 2+ years software development experience
  • 2+ years ML infrastructure experience
  • Experience with data structures/algorithms
  • C++ software system development
  • Performance analysis of compiled ML models
  • Code generation improvement in XLA/Triton compilers
Good to have:
  • 3+ years compiler optimization experience
  • Experience with MLIR
Not hearing back from companies?
Unlock the secrets to a successful job application and accelerate your journey to your next opportunity.

Minimum qualifications:

  • Bachelor's degree in Computer Science, related technical/quantitative field, or equivalent practical experience
  • 2 years of experience with software development in one or more programming languages, and with data structures/algorithms
  • 2 years of experience with ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging)
  • Experience in developing software systems in modern C++

Preferred qualifications:

  • 3 years of experience working on compiler optimization

About the job

Google Cloud's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. 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. As a software engineer, you will work on a specific project critical to Google Cloud's needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. You will anticipate our customer needs and be empowered to act like an owner, take action and innovate. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward.

From startups building their first app to billion-user consumer companies crunching big data, the Google Cloud Platform gives developers and businesses the most advanced data and analytics solution on the market, along with Google's secure, high performance infrastructure. Our team of technical account managers and software engineers work with partners across all functions to develop solutions that harness Google's planet-wide platform to solve the world's most challenging problems.

Responsibilities

  • Profile and analyze the performance of compiled ML models.
  • Improve code generation in the XLA GPU compiler or the Triton compiler.
  • Serve on rotations that triage bugs, guide Public Relations from external contributors, or integrate Multi-Level Intermediate Representation (MLIR) into Google.
View Full Job Description

Add your resume

80%

Upload your resume, increase your shortlisting chances by 80%

About The Company

A problem isn't truly solved until it's solved for all. Googlers build products that help create opportunities for everyone, whether down the street or across the globe. Bring your insight, imagination and a healthy disregard for the impossible. Bring everything that makes you unique. Together, we can build for everyone.

View All Jobs

Get notified when new jobs are added by Google

Similar Jobs

Enphase Energy - Sr. Embedded FW Engineer (Contract)

Enphase Energy, India (On-Site)

Zoox - Manager, Simulation C++ Platform

Zoox, United States (Hybrid)

Playrix - Technical Director (Game Project)

Playrix, Ireland (Remote)

Logitech - Expert Firmware Engineer

Logitech, China (On-Site)

SambaNova Systems - Principal Software Engineer, Runtime

SambaNova Systems, India (Hybrid)

Zuru - Scientific Python Developer

Zuru, Italy (Hybrid)

Nintendo - Embedded Engineer, Display (NTD)

Nintendo, United States (On-Site)

The Walt Disney Company - Staff Production Engineer, Platform-ILM London

The Walt Disney Company, United Kingdom (Hybrid)

Get notifed when new similar jobs are uploaded

Research & Development Jobs

DigitalFish - Senior Computer Vision Engineer, Imaging & Simulation

DigitalFish, United States (On-Site)

Analog Devices - Digital Design Engineer

Analog Devices, India (On-Site)

Cirrus Logic - Analog Design Engineer - Power (PC-64000100)

Cirrus Logic, United Kingdom (Hybrid)

Intel Corporation - Sr. Director, Design Engineering

Intel Corporation, United States (On-Site)

Triomics - Senior Software Engineer

Triomics, India (On-Site)

Spydra - Senior Rust Developer

Spydra, India (On-Site)

Get notifed when new similar jobs are uploaded