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.

• 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.

• 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 practices, and stay updated with the latest advancements in ML and software development.
• Experience with cloud platforms (e.g., AWS, GCP, Azure).
• Familiarity with containerization tools (e.g., Docker, Kubernetes).
• Familiarity with data infrastructures and platforms (e.g., vector databases).
• A track record of contributing to high-quality research projects in deep learning.

What we offer

• The opportunity to contribute to a unique mission with a major impact.
• Comprehensive health benefits.
• A minimum of 20 days vacation per year upon start.