Jobs

Scholarships

Courses

Companies

Placements

Community

Home >

Jobs >

Staff Research Scientist, Machine Learning Efficiency

Staff Research Scientist, Machine Learning Efficiency

1 Hour ago • 4 Years + • Research & Development

About the job

Summary

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.

5 skills required for this role

Test your skills to join the top 1% applicants for this job

cpp

algorithms

prototyping

python

innovation

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

Not hearing back from companies?

Unlock the secrets to a successful job application and accelerate your journey to your next opportunity.

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.

View Full Job Description

About The Company

Google

3628 Active Jobs

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.