Posted in

Software Engineer – Training Infrastructure

Software Engineer – Training Infrastructure

CompanyDeepMind
LocationMountain View, CA, USA
Salary$189000 – $350000
TypeFull-Time
DegreesBachelor’s
Experience LevelSenior

Requirements

  • Bachelor’s degree or equivalent practical experience.
  • 8 years of experience in software development, and with data structures/algorithms.
  • 5 years of experience building, testing, and supporting software in research.
  • Proven track record of building large scale infra for research in Deep Learning, with profound understanding of: Accelerators (e.g. Jax & XLA stack) & performance profiling and optimization, Analysis and debugging of training behavior, Distributed systems, resilience and performance.
  • Experience with Reinforcement Learning a plus.
  • Ability to communicate clearly both verbally and in writing, and comfortable with working in a team distributed across time-zones.

Responsibilities

  • Translate research requirements into technical roadmaps in collaboration with the other teams members.
  • Execute and lead on the implementation and documentation of research infra.
  • Learn about the research problem space the team works in, upskill and be able to contribute to the efforts research agenda.
  • Support growth of more junior team members.
  • Add to the team culture, and be a role model of sustainability and excellence.

Preferred Qualifications

  • Experience with Reinforcement Learning a plus.