Contribute to innovative AI safety initiatives as a Senior Machine Learning Engineer at LawZero in Montreal. Solve complex training and inference challenges using large models in a team-oriented environment. LawZero is looking for a senior distributed machine learning research developer to enhance their focus on AI safety.

You will engage with ML research scientists to tackle problems in model training and inference. Your expertise will be vital in optimizing distributed computing resources and developing essential tools for research experiments. Key Responsibilities:

- Collaborate with researchers to enhance distributed computing for ML
- Investigate performance bottlenecks in research code
- Debug issues and optimize resource utilization




- Develop tools for managing distributed ML experiments
- Document best practices for model development workflows Requirements:
- Degree in computer science, engineering, or related field
- 3+ years designing distributed ML training frameworks
- Effective collaboration skills with cross-functional teams
- Experience with AWS, GCP, or Azure platforms
- Familiarity with GPU profiling tools like PyTorch profiler Make your mark in the field of AI by driving safe machine learning practices at LawZero.

📌 Senior Machine Learning Engineer - LawZero (Montreal)
🏢 LoiZéro
📍 Montreal