Senior Distributed ML Engineer

LawZero is seeking a Senior Distributed ML Engineer to join their AI safety research team. You will collaborate with researchers to solve complex training and inference problems for very large models, optimize distributed computing resources, and develop tools to streamline ML workflows.

• Collaborate with researchers on large-scale distributed ML model training and inference
• Investigate and resolve performance bottlenecks in distributed computing environments
• Develop tools and libraries for orchestrating distributed ML experiments
• Establish best practices for large-scale distributed ML development workflows
• 3+ years of experience in distributed ML training frameworks
• Experience with cloud platforms (AWS, GCP, Azure) and containerization (Docker, Kubernetes)