ML Research Developer

LawZero; Montreal; On-site; CAD 60,000 - 80,000; Full time

Job summary

A non-profit organization in AI research is seeking a Machine Learning Research Developer in Montreal. This role involves collaborating with experts to address challenging ML problems while developing workflows and optimizing resources. Candidates should have relevant degrees, industry experience, and familiarity with tools like PyTorch, TensorFlow, and cloud platforms. An inclusive work environment and generous benefits package are offered.

Benefits

20 days vacation per year

4% retirement savings contribution

Flexible benefits package

Qualifications

• 3+ years of industry experience in complex machine learning workflows.
• Ability to collaborate effectively with cross-functional teams.
• Experience with high performance computing devices.

Responsibilities

• Collaborate with researchers to solve training and inference problems.
• Design and implement workflows for research experiments.
• Develop tools to optimize computing resource usage.

Skills

Machine Learning

PyTorch

TensorFlow

JAX

Cloud platforms

Docker

Kubernetes

Vector databases

Education

Degree in Computer Science

Advanced degree in Machine Learning

Job description

We are seeking a 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.
• 3+ 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.
• 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.

About LawZero

LawZero is a non-profit organization committed to advancing research and creating technical solutions that enable safe-by-design AI systems. Its scientific direction is based on new research and methods proposed by Professor Yoshua Bengio, the most cited AI researcher in the world. Based in Montreal, LawZero’s research aims to build non-agentic AI that learns primarily to understand the world rather than to act in it, giving truthful answers to questions based on transparent and externalized probabilistic reasoning. Such AI systems could be used to accelerate scientific discovery, to provide oversight for agentic AI systems, and to advance the understanding of AI risks and how to avoid them. LawZero believes that AI should be cultivated as a global public good—developed and used safely towards human flourishing. For more information, visit www.lawzero.org

You belong here

At LawZero, diversity is important to us. We value a work environment that is fair, open and respectful of differences. We welcome applications from highly qualified individuals interested in working towards our mission in a respectful, inclusive and collaborative setting.

Your personal information will be collected and processed by LawZero to evaluate your application for employment in compliance with our Privacy Policy. Under privacy laws in force in your country of residence, you may have several privacy rights, such as to request access to your personal information or to request that your personal information be rectified or erased. Details on how you can exercise your rights can be found in our Privacy Policy.

Apply for this job

*

indicates a required field

First Name *

Last Name *

Preferred First Name

Email *

Phone

Country

Phone

Resume/CV *

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Are you legally entitled to work in Canada? * Select...

As part of this recruitment process, you will be required to undergo background checks. Do you wish to proceed? * Select...

Which topic related to AI safety is currently most interesting to you?

Github link

LinkedIn profile

By checking this box, I agree to allow LawZero to store and process my data for the purpose of considering my eligibility regarding my current application for employment. *

By checking this box, I agree to allow LawZero to retain my data for future opportunities for employment for up to 900 days after the conclusion of consideration of my current application for employment.