Senior ML Infra Engineer at General Legal | CA, US | Rezi

Senior ML Infra Engineer at General Legal

Senior ML Infra Engineer

General Legal · CA, US

Yesterday

Senior ML Infra Engineer

General Legal · CA, US

2 days ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now

About the Role

We are 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, making our researchers and engineers dramatically more effective by building reliable infrastructure for model experimentation, evaluation, inference, data generation, training, and observability. 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, with substantial autonomy over architecture and tooling.

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
  • Able to work in-office a minimum of 3 days a week in our SF or NY office locations

Skills

  • Python
  • Cloud infrastructure
  • Distributed systems
  • Containerized workloads
  • Training large language models
  • Fine-tuning large language models
  • Serving large language models
  • GPU infrastructure
  • Distributed training
  • Distributed inference
  • Reinforcement learning
  • Post-training
  • Large-scale evaluation systems
  • AI agents
  • Long-running model workloads
  • Inference latency optimization
  • Inference cost optimization

Location

  • SF
  • NY

Work Type

  • Hybrid

Experience Level

  • Senior

Education Level

  • Computer Science degree or equivalent experience

Salary/Compensations

  • $200,000 - $275,000 base

Benefits

  • Health, dental, vision
  • Unlimited PTO
  • Wellness stipend

About the Company

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