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

A minimum retirement savings employer contribution of 4%.

Generous flexible benefits designed to contribute to your well-being.

A team of passionate experts in their field.

A collaborative and inclusive work environment with offices in the heart of Little Italy, in the trendy Mile-Ex district, close to public transportation.