Senior ML Research Developer

Responsibilities

• Collaborate with researchers to accelerate research, model training and inference
• 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

Requirements

• MSc or higher in machine learning or equivalent work experience
• 5+ years of industry experience in designing and implementing complex machine learning workflows
• Proficiency with PyTorch, TensorFlow, or JAX
• Experience with high performance computing devices
• Experience with cloud platforms (AWS, GCP, or Azure)
• Familiarity with containerization tools like Docker or Kubernetes
• Familiarity with data infrastructures and vector databases
• Track record of contributing to high-quality research projects in deep learning

Benefits

• Comprehensive health benefits
• Minimum of 20 days vacation per year
• Minimum of 4% retirement savings employer contribution
• Generous flexible benefits

About the Company

LawZero is a non-profit organization committed to advancing research and creating technical solutions that enable safe-by-design AI systems. 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.