Senior ML Research Developer

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 work with researchers to accelerate model training, inference, and research implementation. You will design experimental workflows, develop tools and libraries that optimize computing resources, and establish documented best practices for machine learning model development.

Requirements

  • Relevant computer science degree and an advanced degree in machine learning or equivalent work experience
  • 5+ years of industry experience designing and implementing complex ML workflows on high-performance computing devices
  • Experience with PyTorch, TensorFlow, or JAX
  • Experience with AWS, GCP, or Azure
  • Familiarity with Docker and Kubernetes
  • Familiarity with data infrastructure and platforms such as vector databases
  • Track record contributing to deep learning research projects

Responsibilities

  • Collaborate with researchers to accelerate research, model training, and inference
  • Implement novel models and simulated environments efficiently
  • Design workflows for research experiments across small and large-scale projects
  • Develop tools and libraries to optimize computing resources
  • Establish, document, and maintain ML model development best practices

Benefits

  • Comprehensive health benefits
  • At least 20 vacation days per year upon start
  • Minimum 4% employer retirement savings contribution
  • Flexible benefits supporting well-being
Senior ML Research Developer at LawZero | JobStash