Software Engineer, ML Compiler

3 Days ago • 2-5 Years • Research & Development

About the job

Job Description

Google Cloud is seeking a Software Engineer specializing in ML Compiler development. The role involves profiling and analyzing the performance of compiled ML models, improving code generation within XLA GPU or Triton compilers, and contributing to bug triaging, public relations, and MLIR integration. This position requires at least 2 years of software development experience (C++ preferred), data structures/algorithms knowledge, and ML infrastructure expertise (model deployment, evaluation, data processing). Responsibilities include optimizing compiler performance, improving code generation efficiency, and collaborating on various compiler-related projects within Google's fast-paced environment.
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
  • Modern C++ development skills
Good to have:
  • 3+ years compiler optimization experience
  • Experience with XLA GPU or Triton compiler

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.

San Francisco, California, United States (On-Site)

Mountain View, California, United States (On-Site)

Warsaw, Masovian Voivodeship, Poland (On-Site)

San Bruno, California, United States (On-Site)

Mexico City, Mexico City, Mexico (On-Site)

Dublin, County Dublin, Ireland (On-Site)

Hyderabad, Telangana, India (On-Site)

View All Jobs

Get notified when new jobs are added by Google

Similar Jobs

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Trailmix Games - Analytics Engineer

Trailmix Games, United Kingdom (Hybrid)

GapBridge - Android Platform developer

GapBridge, India (Hybrid)

Keywords Studios (Player Support) - Game Engine Programmer

Keywords Studios (Player Support), Japan (Remote)

DNEG - Facial Rigger

DNEG, Canada (On-Site)

Playrix - Senior C++ Software Engineer (Gameplay)

Playrix, Kazakhstan (Remote)

Embark Studios - Online Game Programmer (C++) - Game Services

Embark Studios, Sweden (On-Site)

Meta - Software Engineering Manager, Product

Meta, United States (On-Site)

Rockstar Games - Associate Principal Technical Artist: Performance Capture

Rockstar Games, United States (On-Site)

Get notifed when new similar jobs are uploaded

Jobs in Munich, Bavaria, Germany

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Get notifed when new similar jobs are uploaded