Research Scientist, People and AI Research (PAIR)

Key Skills

AJAX, JAX framework, Mandarin Chinese Language, NeurIPS, Oracle Analytics, T5X framework

Job Description

Minimum qualifications:

PhD in Computer Science, Artificial Intelligence, Machine Learning, a related technical field, or equivalent practical experience.

Experience writing machine learning code, prototyping with machine learning models and working with transformer models.

Preferred qualifications:

• Experience working in an industry or academic research lab, and taking research from concept to products.
• Experience using and advanced interpretability methods such as mechanistic interpretability, training data attribution or saliency.
• Experience in research alignment and control of machine learning systems such as few short learning or parameter efficient methods.
• Experience with symbolic computation and problem-complexity such as automated reasoning or complexity theory.
• Contributions to publications related to machine learning problem complexity (e.g., NeurIPS, ICLR, ICML, CHI, EMNLP, ACL).

About the job

People and AI Research (PAIR) is a team in Google Research focused on Human-centered research and design to make AI partnerships productive. Our multidisciplinary team does fundamental research, invents new technology, and creates frameworks for design in order to drive a human-centered approach to AI.

In this role, you will be conducting research state-of-the-art on small problems, Machine Learning algorithms, interpretability, and visualization to better understand and control machine learning models, leverage small datasets, and improve applications of larger language-driven models. You will write Machine Learning code, coordinate with other teams, and develop, use, and contribute to emerging software systems and work with small and large language models. You will also work with others in PAIR and Google to explore new forms of interaction that can support the alignment of machine learning with human values.

Google Research addresses challenges that define the technology of today and tomorrow. From conducting fundamental research to influencing product development, our research teams have the opportunity to impact technology used by billions of people every day.

Our teams aspire to make discoveries that impact everyone, and core to our approach is sharing our research and tools to fuel progress in the field -- we publish regularly in academic journals, release projects as open source, and apply research to Google products.

Responsibilities

• Lead advanced research on small problems and interpretability to develop new algorithm insights for new alignable Machine Learning architectures.
• Write Machine Learning code in frameworks like JAX, T5X, TFJS, etc.
• Coordinate research efforts within PAIR and outside, manage objectives and priorities, and drive ambitious and thoughtful research directions towards alignable neural algorithms.
• Present and disseminate research internally and externally, e.g. by contributing to interactive explorable visualizations (e.g., PAIR explorables), academic papers, and blog posts.