Research Scientist, AI for Code

1 day ago
Full time role
Remote · New York, NY, US... more

At Google DeepMind, we value diversity of experience, knowledge, backgrounds and perspectives and harness these qualities to create extraordinary impact. We are committed to equal employment opportunity regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, pregnancy, or related condition (including breastfeeding) or any other basis as protected by applicable law. If you have a disability or additional need that requires accommodation, please do not hesitate to let us know.

Snapshot

Artificial Intelligence could be one of humanity’s most useful inventions. At Google DeepMind, we’re a team of scientists, engineers, machine learning experts and more, working together to advance the state of the art in artificial intelligence. We use our technologies for widespread public benefit and scientific discovery, and collaborate with others on critical challenges, ensuring safety and ethics are the highest priority.

About Us

Artificial Intelligence could be one of humanity’s most useful inventions. At Google DeepMind, we’re a team of scientists, engineers, machine learning experts and more, working together to advance the state of the art in artificial intelligence. We use our technologies for widespread public benefit and scientific discovery, and collaborate with others on critical challenges, ensuring safety and ethics are the highest priority.

The Role

Key Responsibilities:

You’ll be improving the most promising models and research as it relates to high-impact real world problems, by rapid-prototyping initial concepts and designing and running experiments. You will gain an understanding of the path from research to production whilst working on real world problems, with the chance to see your work make a difference. You’ll be embedded within a small project team of  Research Engineers, and will be expected to drive the foundational research direction with the potential for significant impact in the short and long term.

Some of our current projects are focusing on the areas below so experience in any of these is highly beneficial:

  • Code Generation and Understanding
  • Natural Language Processing
  • Multimodal Understanding
  • Reinforcement Learning
  • Agentic Workflows

About You

We seek out individuals who thrive in ambiguity and who are willing to help out with whatever moves prototypes forward. We regularly need to invent novel solutions to problems, and often change course if our ideas don’t work out, so flexibility and adaptability to work on any project is a must.

In order to set you up for success as an AI for Code Research Scientist at Google DeepMind, we look for the following skills and experience:

  • MSc or PhD/DPhil degree in computer science, mathematics, applied stats, machine learning or similar experience working in industry
  • Proven knowledge and experience of Python or C++

Experience in AI for Code, such as prior research and publications

  • Deep foundational knowledge of machine learning and statistics.
  • Proven experience of JAX or similar ML frameworks (e.g. TensorFlow, PyTorch ) is highly desirable.
  • Experience pretraining, fine-tuning large, and/or applying RL to models.
  • Experience applying and productionizing state-of-the-art large visual, language and multimodal research.
  • Software Engineering experience and experience working on large-scale ML projects highly desirable.
  • Proven experience working in industry, working on projects from proof-of-concept through to implementation highly beneficial.
  • A passion for Artificial Intelligence.
  • Great communication skills and proven interpersonal skills.

In addition, the following would be an advantage:

  • Experience in applying experimental ideas to applied problems.
  • Cross functional collaboration experience.
  • Strong publication record in top machine learning conferences (e.g., NeurIPS, CVPR, ICML, ICLR, ICCV, ECCV).
  • Prior experience working with product teams.

The US base salary range for this full-time position is between $136,000 - $210,000 + bonus + equity + benefits. Your recruiter can share more about the specific salary range for your targeted location during the hiring process.