Senior Distributed ML Engineer
LawZero · Berlin · Montreal
Research instituteonsitesenior
At a glance
Highlights
- Work on novel AI safety agenda guided by Yoshua Bengio's research.
- Solve large-model training and inference challenges alongside research scientists.
- Build safe-by-design AI for scientific discovery and agentic oversight.
Ideal candidate
Collaborative systems-minded engineer passionate about safe-by-design AI, energized by accelerating research and optimizing large-scale model workflows alongside scientists.
Requirements
- Degree in computer science or related field; advanced ML degree preferred
- 3+ years designing and implementing distributed ML training frameworks
- Recent experience with Megatron, DeepSpeed, HuggingFace Accelerate, FSDP, vLLM and/or verl
- Experience with cloud platforms (AWS, GCP, Azure) and workload managers (Ray, SLURM)
- Experience with GPU profiling tools such as PyTorch profiler, PyProf, NVIDIA Nsight
- Familiarity with containerization tools and data infrastructures including vector databases
- Track record contributing to high-quality deep learning research projects
Lab profile
About LawZero
Nonprofit building safe-by-design AI systems including Scientist AI, led by Yoshua Bengio.
What they build
- Scientist AI models
- Safety guardrails for frontier LLMs
Known for
- Safe-by-design AI research
- Scientist AI approach
- Rigorous capability and safety evaluations
Key facts
- Founded in 2025
- Headquartered in Montréal, Canada
- Nonprofit funded by private foundations and public grants
Posted · Last seen