Machine Learning Research Scientist in Probabilistic Inference

Montréal-based nonprofit AI-safety research organization developing safe-by-design advanced AI, centered on its Scientist AI approach.

Recently fundedCompany intelligence
Montréal, Québec, Canada
About LawZero

LawZero develops technical solutions and research for highly capable, non-agentic, safe-by-design AI systems. Its Scientist AI research aims to provide transparent, evidence-based prediction and potential oversight/guardrails for agentic AI.

View jobs by LawZero

Skills

About the Role

You will develop amortized inference and learning methods for probabilistic graphical models. You will design evaluations, collaborate on theory, implement research in Python and machine learning frameworks, analyze experimental results, and communicate findings to stakeholders.

Requirements

  • Advanced degree in computer science, mathematics, or a relevant field
  • At least 3 years of deep learning research experience
  • Expertise in probabilistic inference
  • Strong mathematics background
  • Experience developing and implementing machine learning models
  • Python proficiency
  • Experience with PyTorch or TensorFlow
  • Analytical, problem-solving, and communication skills
  • Research contributions in probabilistic inference or related fields

Responsibilities

  • Develop amortized inference methods for high-dimensional distributions
  • Develop parameter- and structure-learning methods for probabilistic graphical models
  • Design evaluation strategies for probabilistic inference methods
  • Collaborate on learning and inference theory
  • Translate theoretical proposals into high-quality implementations
  • Analyze experimental results to guide research
  • Communicate research findings to stakeholders

Benefits

  • Comprehensive health benefits
  • Minimum 20 days of vacation per year upon start
  • Minimum 4% employer retirement savings contribution
  • Flexible benefits for well-being
Machine Learning Research Scientist in Probabilistic Inference at LawZero | JobStash