ML Developer (Applied Research)

A degree in a relevant computer science field (e.g., computer science, computer engineering, software engineering) is required, along with an advanced degree (MSc or higher) in machine learning or equivalent work experience.

3+ years of industry experience in designing and implementing complex machine learning workflows on high performance computing devices using PyTorch, TensorFlow, or JAX.

Ability to collaborate effectively with cross-functional teams, document best practices, and stay updated with the latest advancements in ML and software development.

Experience with cloud platforms (e.g., AWS, GCP, Azure) and workload managers (e.g., Ray, SLURM)

Familiarity with containerization tools (e.g., gRPC, Docker, Kubernetes).

Familiarity with data infrastructures and platforms (e.g., vector databases).

The opportunity to contribute to a unique mission with a major impact.

Comprehensive health benefits.

A minimum of 20 days vacation per year upon start.

A minimum retirement savings employer contribution of 4%.

Generous flexible benefits designed to contribute to your well-being.

A team of passionate experts in their field.

Accelerate applied research, model training and inference, and iterate on implementing models for real-world applications (that will form the basis of LawZero’s future solutions offerings).

Design and implement workflows for research experiments across simulated environments and real-world applications.

Develop datasets, tooling, dashboards, and libraries to adapt, monitor, interpret, and evaluate models in the context of real-world applications.

Establish, document, and maintain best practices for ML model development workflows.

Redesign and adapt research ideas and prototypes into robust production-grade tools and solutions.

Deeply understand customer use-cases to inform training strategies and surface edge cases.