ML Developer (Applied Research)

We are seeking a machine learning (ML) developer (applied research) to join our team working on the Scientist AI. In this role, you will work closely with ML research scientists and product stakeholders to start applying the Scientist AI to real-world problems.

Key responsibilities

• 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.
  
  

Skills and qualifications

• 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).
• Experience building new ML products from scratch, or based on research prototypes.
• Experience working alongside researchers is a bonus.

What we offer

• 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.
• A collaborative and inclusive work environment with offices in the heart of Little Italy, in the trendy Mile-Ex district, close to public transportation.