Harvey Logo

Harvey

Research Engineer, Infrastructure

Posted 7 Days Ago
Be an Early Applicant
Hybrid
San Francisco, CA, USA
230K-450K Annually
Entry level
Hybrid
San Francisco, CA, USA
230K-450K Annually
Entry level
Build and scale infrastructure for post-training open-weight AI models, including distributed training, RL rollouts, agent environments, evaluation systems, experiment tracking, model serving, and production integration. Own performance, reliability, GPU utilization, checkpointing, failure recovery, observability, and bottleneck removal. Partner closely with researchers to turn experimental workflows into dependable systems.
The summary above was generated by AI
Why Harvey

At Harvey, we’re transforming how legal and professional services operate. By combining frontier agentic AI, an enterprise-grade platform, and deep domain expertise, we’re reshaping how critical knowledge work gets done for decades to come.

This is a rare chance to help build a generational company at a true inflection point. We have strong product-market fit and world-class investor support. We’re scaling fast and defining a new category in real time. The work is ambitious, the bar is high, and the opportunity for growth — personal, professional, and financial — is unmatched.

Our team moves fast, takes ownership, and is deeply committed to the mission — operating with intensity, staying close to our customers, and pushing each other for excellence. We live by three values: Decisiveness, Simplicity, and Job's Not Finished. We act quickly on clear judgment over perfect information, we believe simplicity is what scales, and we're never satisfied with where we are. If you want to do the best work of your career alongside people who share that drive, we'd love to build with you.

At Harvey, the future of professional services is being written today — and we’re just getting started.

Role Overview

In this role, you'll build the systems that power how we adapt and improve leading open-weight models for legal work.

We recently post-trained our first model built for legal work, and we’re now building the team to scale up our post-training efforts. You’ll own the infrastructure behind that effort: distributed training, RL rollout systems, the environments where models learn to use tools and complete complex legal workflows, and the internal systems the team uses to manage data, run evals, and launch and track training jobs. You’ll help decide what we build, what we adopt, and how the pieces fit together. You’ll join a small, high-bar team of hands-on engineers and researchers, where each person owns a large part of the stack.

We’re looking for research engineers who have built demanding ML or distributed systems, can debug across the stack, and want to work closely with researchers to make training faster, more reliable, and easier to iterate on.

Learn more about our work at harvey.ai/research.

What You'll Do
  • Scale our post-training stack, supporting SFT and RL on leading open-weight models.

  • Build distributed rollout infrastructure that coordinates model inference, tool execution, and collection of training experience for long-horizon tasks.

  • Build the infrastructure for reproducible agent environments where models can interact with documents, tools, and external systems during training and evaluation.

  • Own the performance and reliability of training runs, including GPU utilization, checkpointing, failure recovery, and observability.

  • Identify and remove bottlenecks across training and inference to shorten the path from research idea to useful result.

  • Build the internal tools and services the team runs on: training data management and versioning, running evals and browsing results, and launching and tracking training jobs and experiments.

  • Integrate models into production, including serving, latency, cost, and online evaluation, so research reaches customers quickly and safely.

  • Partner closely with researchers to turn experimental workflows into dependable systems while keeping room for rapid experimentation.

What You Have
  • Experience building ML infrastructure, distributed systems, or research tooling, with ownership of performance, reliability, and operation.

  • Strong Python and generalist software engineering skills across backend services, data systems, and internal tools, with familiarity with PyTorch, JAX, or similar ML frameworks.

  • Experience with distributed training or inference, including diagnosing performance bottlenecks and failures across the stack.

  • Understanding of post-training workflows; experience with RL rollouts, agent environments, or sandboxed tool execution is a plus.

  • Practical judgment about what to build versus adopt, and the ability to partner with researchers and ship quickly as needs evolve.

We are hiring for this role across multiple levels.

Compensation

$230,000 - $450,000 USD

Depending on your location, an Applicant Privacy Notice may apply to you. You can find all of our Applicant Privacy Notices here.

#LI-DZ1

Harvey is an equal opportunity employer and does not discriminate on the basis of race, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition, or any other basis protected by law.

We are committed to providing reasonable accommodations to applicants with disabilities, and requests can be made by emailing [email protected]

HQ

Harvey San Francisco, California, USA Office

San Francisco, California, United States

Similar Jobs

3 Days Ago
In-Office
Palo Alto, CA, USA
150K-220K Annually
Entry level
150K-220K Annually
Entry level
Artificial Intelligence • Machine Learning • Software
Build and own ML infrastructure for distributed training across GPU clusters, including job launching, checkpointing, recovery, monitoring, profiling, experiment tooling, and reliable data pipelines. Debug failures across GPUs, networking, numerics, and data while partnering closely with research. The role suits an early-career builder with strong Python skills, practical ML tooling experience, public work, high agency, and a willingness to learn quickly in a small, autonomous team.
Top Skills: CheckpointingData PipelinesDistributed TrainingExperiment TrackingGpu ClustersMachine Learning ToolingMonitoringNetworkingPython
14 Days Ago
In-Office or Remote
3 Locations
105K-260K Annually
Senior level
105K-260K Annually
Senior level
Artificial Intelligence • Information Technology • Software
Build reliable software and infrastructure for frontier AI agent research, including environment execution, trajectory collection, evaluations, experiment orchestration, data pipelines, and observability. Partner with research engineers to turn prototypes into reusable, tested systems; improve reproducibility, scalability, cost efficiency, and recovery from failed runs; and own infrastructure from design through operation.
Top Skills: APIsCloud InfrastructureContainerized EnvironmentsData PipelinesDistributed Job ExecutionDockerLinuxPythonReinforcement Learning
18 Days Ago
In-Office or Remote
San Francisco, CA, USA
150K-350K Annually
Entry level
150K-350K Annually
Entry level
Artificial Intelligence • Software
Build and optimize infrastructure for frontier-scale reinforcement learning and distributed model training. Responsibilities include improving compute, memory, networking, scheduling, kernels, communication paths, runtimes, and parallel training systems. The role contributes to RL training architecture, asynchronous rollouts, post-training systems, open-source ML infrastructure, and GPU performance optimization while collaborating with research and infrastructure teams.
Top Skills: CudaDeepspeedFsdpGpu ArchitectureHigh-Performance NetworkingMegatronMulti-Node Gpu ClustersPyTorchPytorch DistributedRayTritonVllm

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account