Staff Research Scientist, Machine Learning Efficiency

3 Months ago • 4 Years + • Research & Development

Job Summary

Job Description

As a Staff Research Scientist in Machine Learning Efficiency at Google, you will lead efforts to improve the efficiency of large language models. This involves designing and implementing novel ML algorithms, running and analyzing experiments, and collaborating with hardware engineers to co-design high-performance models and next-generation ML accelerators (TPUs). Responsibilities include generating innovative solutions, defining long-term research agendas, collaborating with global research teams, and publishing findings. The role requires a PhD in Computer Science or a related field, 4+ years of experience in ML efficiency research, proficiency in programming languages (Python, C/C++), and experience with experimental model implementation and analysis. You will contribute to Google's portfolio of research projects, impacting product development and infrastructure goals.
Must have:
  • PhD in CS or related field
  • 4+ years ML efficiency research experience
  • Programming (Python, C/C++)
  • Experimental model implementation & analysis
  • Publication record (ICML, ICLR, NeurIPS)
  • Lead research team
  • Collaborate globally
Good to have:
  • Innovative research experience
  • Hardware capabilities & constraints knowledge
  • Chip design understanding

Job Details


Minimum qualifications:

  • PhD in Computer Science, a related technical field, or equivalent practical experience.
  • 4 years of experience with research agendas across multiple teams or projects in Machine Learning (ML) efficiency, optimization, or a related field.
  • Experience with programming languages (e.g., Python, C or C++).
  • Experience implementing, running, analyzing experimental models or transformer architecture.
  • One or more scientific publication submissions for conferences, journals, or public repositories (e.g., ICML, ICLR, NeurIPS).

Preferred qualifications:

  • Experience in innovative research.
  • Experience as a leader within a research team.
  • Ability to effectively navigate ambiguity.
  • Knowledge of existing hardware capabilities and constraints.
  • Understanding of chip design.

About the job

As an organization, Google maintains a portfolio of research projects driven by fundamental research, new product innovation, product contribution and infrastructure goals, while providing individuals and teams the freedom to emphasize specific types of work. As a Research Scientist, you'll setup large-scale tests and deploy promising ideas quickly and broadly, managing deadlines and deliverables while applying the latest theories to develop new and improved products, processes, or technologies. From creating experiments and prototyping implementations to designing new architectures, our research scientists work on real-world problems that span the breadth of computer science, such as machine (and deep) learning, data mining, natural language processing, hardware and software performance analysis, improving compilers for mobile platforms, as well as core search and much more.

As a Research Scientist, you'll also actively contribute to the wider research community by sharing and publishing your findings, with ideas inspired by internal projects as well as from collaborations with research programs at partner universities and technical institutes all over the world.

In this role, you will lead our Machine Learning (ML) efficiency effort. You will work on next-generation large language models architecture, in close partnership with hardware engineers and researchers. Recent breakthroughs in large language models performance led to many new research directions, and a lot of teams are focusing on maximizing performance when using the existing hardware. There are also teams designing the next generation of hardware based on an extrapolation of the current modeling efforts. You will co-design high-performance models along with the next generation of ML accelerators (TPUs) that can power them efficiently.

Google Research is building the next generation of intelligent systems for all Google products. To achieve this, we’re working on projects that utilize the latest computer science techniques developed by skilled software developers and research scientists. Google Research teams collaborate closely with other teams across Google, maintaining the flexibility and versatility required to adapt new projects and foci that meet the demands of the world's fast-paced business needs.

Responsibilities

  • Generate, design ideas, leading a team for finding innovative solutions through theoretical or empirical insights.
  • Design and implementing Machine Learning (ML) algorithms, running and analyzing experiments, incorporating components that could constitute part of a future Tensor Processing Unit (TPU).
  • Collaborate with research teams located across the globe.
  • Define long-term research agendas, intermediate milestones, and align with research and product partners.
  • Impact through publication and scientific dissemination.

Similar Jobs

CD PROJEKT RED - Expert Rendering Engineer

CD PROJEKT RED

Boston, Massachusetts, United States (Hybrid)
4 Months ago
Meta - Software Engineer, Pathways Program

Meta

San Francisco, California, United States (On-Site)
3 Months ago
Keywords Studios (Player Support) - Lead Programmer

Keywords Studios (Player Support)

(Hybrid)
2 Months ago
Google - Senior Technical Solutions Consultant, Extended Workforce Solutions, gTech

Google

Toronto, Ontario, Canada (On-Site)
2 Months ago
PlayStation Global - Programmeur·euse Senior – Jouabilité/Senior Gameplay Programmer

PlayStation Global

Montreal, Quebec, Canada (On-Site)
2 Months ago
ByteDance - Software Engineer in Large Model System Graduate (Machine Learning Sys-US) - 2024 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
ByteDance - LLM Software Engineer/Researcher Graduate (Applied Machine Learning) - 2024 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Intel Corporation - Senior Logic Design Verification Engineer

Intel Corporation

Penang, Malaysia (Hybrid)
2 Months ago
Fluence - Controls Software Engineer II

Fluence

Houston, Texas, United States (Hybrid)
4 Months ago
Matic Robots - Systems  Engineer (Embedded Linux)

Matic Robots

Canada (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Backend Software Engineer, Insights, Benchmarks, and Narratives

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
2 Months ago
Google - Senior Software Engineer, Google Cloud Data Management

Google

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
2 Months ago
ByteDance - Software Engineer (Data Security-US) - 2025 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Google - Senior Quantitative UX Researcher, Core

Google

New York, New York, United States (On-Site)
2 Months ago
MADFINGER Games - Senior Gameplay Programmer - Unreal

MADFINGER Games

Brno, South Moravian Region, Czechia (On-Site)
3 Months ago
CD PROJEKT RED - Senior Engineer, Graphics

CD PROJEKT RED

Warsaw, Masovian Voivodeship, Poland (On-Site)
7 Months ago
Google - Software Engineer, Machine Learning, Google Cloud

Google

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Tencent - Game AI Engineer Intern 104492

Tencent

London, England, United Kingdom (On-Site)
2 Months ago
Wind River Systems - Senior Cloud Solutions Architect

Wind River Systems

Singapore, Singapore (On-Site)
4 Months ago
Zynga - Engineering Manager

Zynga

Bengaluru, Karnataka, India (Hybrid)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Worldwide

Looks like we're out of matches

Set up an alert and we'll send you similar jobs the moment they appear!

Research & Development Jobs

Sony Interactive Entertainment - PlayStation向けシステム・ネットワーク技術R&D ソフトウェアエンジニア

Sony Interactive Entertainment

Tokyo, Japan (Hybrid)
4 Months ago
Google - Power Engineer, Platforms, Engineering

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
Razer - Software Engineering Manager

Razer

Shah Alam, Selangor, Malaysia (On-Site)
4 Months ago
Whirlpool Corporation - Senior Lead SW Engineer

Whirlpool Corporation

Pune, Maharashtra, India (On-Site)
4 Months ago
Intel Corporation - Pre-Si Verification Lead/Manger

Intel Corporation

Bengaluru, Karnataka, India (Hybrid)
3 Months ago
Google - Senior SoC and IP Design Engineer, Google Cloud

Google

(On-Site)
3 Months ago
ByteDance - Software Engineer - Data Transmission- San Jose

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Microsoft - Technical Support Engineering

Microsoft

Bengaluru, Karnataka, India (Hybrid)
2 Months ago
Google - Lead Graphics Processing Unit Architect

Google

Mountain View, California, United States (On-Site)
2 Months ago
Netflix - Engineering Manager, DevEx

Netflix

United States (Remote)
2 Months ago

Get notifed when new similar jobs are uploaded

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

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug