Senior ML Research Developer

LawZero is a non-profit organization committed to advancing research and creating technical solutions that enable safe-by-design AI systems. They are seeking a senior machine learning research developer to work closely with ML research scientists on a novel AI safety agenda, solving complex training and inference problems using large models.

Responsibilities:

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