ML Infra Engineer

The Opportunity

The legal industry is broken — lawyers bill by the hour, charge enormous sums, and take forever to respond. This means that a lot of people who need legal advice don't get it. We're using AI to solve this problem. AI takes the first pass at everything our lawyers do, so that the lawyers can focus on human interaction. We're starting with negotiating commercial contracts, and will expand quickly from there to other use cases within legal. We're reimagining what a contract can be in the AI age. We're a veteran team, with one successful exit already, who've been doing deep learning in the legal space since long before ChatGPT.

Description

General Legal is seeking a Senior ML Infra Engineer to build the systems that allow us to rapidly experiment with, evaluate, train, and deploy increasingly capable AI systems.

You'll sit at the intersection of research and production engineering. Your job will be to make our researchers and engineers dramatically more effective: building reliable infrastructure for model experimentation, evaluation, inference, data generation, training, and observability while ensuring that promising ideas can move quickly from an experiment into production.

This is a unique opportunity to join us at the ground level and define the AI infrastructure behind a platform that will set new standards for the legal industry. You'll have substantial autonomy over architecture and tooling and will help determine how our ML stack evolves as we scale.

Responsibilities

• Build and own infrastructure for training, evaluating, and serving AI models
• Develop systems for running large-scale experiments and evaluations quickly and reproducibly
• Build pipelines for generating, processing, versioning, and managing training and evaluation data
• Improve the reliability, latency, throughput, and cost efficiency of model inference
• Build observability and monitoring for AI systems in production
• Develop infrastructure for agentic workloads, including long-running and asynchronous model execution
• Work closely with research scientists and product engineers to turn new AI capabilities into reliable production systems
• Evaluate and integrate new models, inference systems, training frameworks, and infrastructure as the field evolves
• Be proactive and come up with ideas on how to build our AI systems better

Requirements

• 5+ years of software engineering, machine learning engineering, or infrastructure experience
• Strong software engineering fundamentals and proficiency in Python
• Experience building production infrastructure for machine learning systems
• Experience with cloud infrastructure, distributed systems, and containerized workloads
• Ability to independently design, build, and operate complex technical systems
• Computer Science degree or equivalent experience
• Eligible to work in the US, and able to work in-office a minimum of 3 days a week in our SF or NY office locations.

Nice-to-Haves

• Experience training, fine-tuning, or serving large language models
• Experience with GPU infrastructure and distributed training or inference
• Experience with reinforcement learning, post-training, or large-scale evaluation systems
• Experience building infrastructure for AI agents or long-running model workloads
• Experience optimizing inference latency and cost at scale
• Experience at an early-stage startup

Compensation & Benefits

$200,000 - $275,000 base, calibrated to experience and location

Health, dental, vision; unlimited PTO; wellness stipend & more…