Senior ML Research Scientist (probabilistic inference) Montreal

LawZero; Montreal; On-site; CAD 100,000 - 150,000; Full time

Job summary

Join LawZero as a Senior ML Research Scientist, where you will lead research on AI safety. In this role, you will focus on developing and evaluating advanced probabilistic inference methods, collaborating with experts while translating theoretical insights into practical implementations. Enjoy a competitive salary, generous vacation, and the opportunity to work in a vibrant team dedicated to safe AI development.

Benefits

Minimum 20 days vacation per year

4% employer contribution to retirement savings

Generous flexible benefits

Collaborative work environment in Little Italy

Qualifications

• PhD preferred; strong mathematical background.
• Minimum 3 years of experience in deep learning research.
• Track record of high-quality research in probabilistic inference.

Responsibilities

• Develop and evaluate probabilistic inference methods.
• Collaborate with mathematicians on relevant research.
• Communicate complex findings effectively to stakeholders.

Skills

Expertise in probabilistic inference

Amortized inference methods

Analytical skills

Problem-solving skills

Communication skills

Education

Advanced degree in Computer Science or Mathematics

Tools

Python

PyTorch

TensorFlow

Job description

Senior ML Research Scientist (probabilistic inference)

Montreal

We are seeking a senior machine learning (ML) researcher to join our team working on a novel AI safety research agenda. In this role, you will develop and evaluate probabilistic inference methods, with a focus on amortized inference, translating theoretical insights into practical implementations.

Key responsibilities

• Develop amortized inference methods suitable for high-dimensional discrete and continuous distributions.
• Develop parameter- and structure-learning methods for large probabilistic graphical models that benefit from amortized probabilistic inference.
• Design evaluation strategies for GS methods that rely on probabilistic inference.
• Collaborate with mathematicians on theory related to learning and inference in probabilistic models.
• Translate theoretical proposals into high quality implementations in a programming language such as Python.
• Analyze and interpret experimental results to steer future research directions.
• Communicate complex findings effectively to various stakeholders.

Skills and qualifications

• Advanced degree in a relevant field (e.g., Computer Science, Mathematics). A PhD is preferred but not required if the candidate demonstrates exceptional abilities.
• A minimum of 3 years of experience in deep learning research.
• Expertise in probabilistic inference is required, in addition to expertise in one or more of the following:
  • Amortized inference methods (including GFlowNets)
  • Sampling-based approximate inference methods
  • Bayesian inference
  • Parameter- and/or structure-learning in probabilistic graphical models (including causal models)
• Strong background in mathematics.
• Proven experience in developing and implementing machine learning models.
• Proficiency in programming languages such as Python, and experience with ML frameworks like PyTorch or TensorFlow.
• Excellent analytical and problem-solving skills, with a demonstrated ability to think critically about complex systems.
• Strong communication skills, both written and verbal, with the ability to explain complex ideas to diverse audiences.
• Track record of contributing to high-quality research in probabilistic inference or related fields.
• Ability to work collaboratively in a team environment while also being self-motivated and independent.

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

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.