Senior ML Research Developer

We are seeking a senior machine learning (ML) research developer to join our team working on a novel AI safety agenda. In this role, you will work closely with ML research scientists to solve difficult training and inference problems using very large models.

Key responsibilities

• Collaborate with researchers to accelerate research, model training and inference, and facilitate the efficient implementation of novel models and simulated environments.
• Design and implement workflows for research experiments across toy scenarios and large scale projects.
• Develop tools and libraries to optimize the use of computing resources.
• Establish, document, and maintain best practices for ML model development workflows.

Skills and qualifications

• A degree in a relevant computer science field (e.g., computer science, computer engineering, software engineering) is required, along with an advanced degree (MSc or higher) in machine learning or equivalent work experience.
• 5+ years of industry experience in designing and implementing complex machine learning workflows on high performance computing devices using PyTorch, TensorFlow, or JAX.
• Ability to collaborate effectively with cross-functional teams, document best pract