Full-time

Machine Learning Developer for AI Safety

Posted by LawZero • June 01, 2026

📍 montreal (administrative region), qc, Canada
Apply Now

Description

Enhance AI safety work as a Machine Learning Developer specializing in distributed systems. Partner with researchers to solve training challenges and optimize large model inferences efficiently.
This senior position focuses on addressing complex performance issues and developing essential tools for large-scale ML projects. You will also help document and implement best practices for the successful development of distributed machine learning workflows. Your impact will be critical in driving pioneering AI safety research efforts.
Key Responsibilities:
• Accelerate ML research by collaborating with scientists
• Identify and debug performance bottlenecks in code
• Build tools for effective management of distributed resources
• Maintain and document best practices for ML workflows
Requirements:
• Degree in a related computer science discipline
• Minimum of 3 years designing distributed ML frameworks
• Familiarity with cloud platforms and workload managers
•...

Ready to Seal the Deal?

Submit your application today and take the next step in your career with LawZero.

Apply for this Job