Research Engineer – Training and Environment Infrastructure

MaxIT Consulting - Max Corporate Group; San Francisco (CA); Hybrid; USD 120,000 - 180,000; Full time

Benefits offered by this job

Health, dental and vision coverage

Hybrid work arrangement

Job summary

MaxIT Consulting – Max Corporate Group seeks a Research Engineer to build the training and environment infrastructure for a fast-growing AI company in San Francisco. You will develop scalable systems to support long-horizon agent evaluation, training and deployment pipelines with real enterprise relevance.

The role sits on a small founding team and offers hybrid work in SF, a permanent position, and mid-level responsibilities with ASAP start.

Qualifications

• 1–5 years of full-time experience (at least one year).
• Deep experience in distributed systems, containers/OS internals, ML infrastructure or model training.
• Experience with training or serving models at scale or building environments/benchmarks.
• Publication at a top venue (ICLR or NeurIPS).
• Built infrastructure that others depend on beyond a single research script.
• Professional English for daily technical collaboration.

Responsibilities

• Build checkpoint and restore of full machine state, including memory and GPU state, across many machines.
• Develop the environment orchestrator for thousands of environments in a multi-GPU cluster.
• Provision real cloud accounts for agents, with hundreds created/deleted on demand.
• Create realistic simulated companies for training and evaluation.
• Serve as the infrastructure foundation for research and deployed teams.

Skills

Distributed systems

Containers & OS internals

ML infrastructure

Model training

Education

Master's degree in CS or related field

Tools

Checkpointing

Cluster orchestration

Scheduling large jobs

Job description

Research Engineer – Training and Environment Infrastructure

San Francisco, CA | Hybrid | Full-Time | Permanent Employment | Mid-Level | ASAP Start

MaxIT Consulting – Max Corporate Group is looking for a Research Engineer to build the training and environment infrastructure of a fast-growing AI company in San Francisco. The company builds evaluation environments and post-training infrastructure for long-horizon enterprise agents, and uses that stack to train and deploy custom models for very large technology companies.

It is a small, founding-stage team out of top AI labs and technology companies, already profitable and with real enterprise revenue. This role sits under the whole company: the research and deployed teams build directly on what you own.

THE CHALLENGE

As agent workflows get longer, evaluation and training infrastructure become the bottleneck. You will own the systems that let long runs be paused, rewound and continued, and that start, watch and shut down thousands of heterogeneous environments cheaply and reliably.

Within your first 90 days you will own one of the core systems end to end: checkpoint and restore, the environment orchestrator, or the simulated companies used for training and evaluation.

KEY RESPONSIBILITIES

• Build checkpoint and restore of full machine state, including memory and GPU state, across many machines.
• Build the orchestrator that starts, monitors and shuts down thousands of environments, from a single container to a multi-GPU cluster.
• Provision real cloud accounts for agents, with hundreds created and deleted on demand.
• Build realistic simulated companies with code, people, tickets, threads and documentation for agent training and evaluation.
• Be the infrastructure foundation that the research and deployed teams depend on.

WHAT WE ARE LOOKING FOR

• One to five years of full-time experience (at least one year full time).
• Deep experience in at least one of: distributed systems, containers and operating-system internals, ML infrastructure or model training.
• Has trained or served models at real scale (distributed training, inference or serving), or built the environment or benchmark behind a published result.
• A publication at a top venue (such as ICLR or NeurIPS).
• Has built and owned infrastructure that other people depended on, beyond a single research script.
• Comfortable choosing the right tools: no specific language or framework is required.
• Professional English, as the role works daily with technical teams and enterprise customers.

NICE TO HAVE

• Experience with checkpointing, cluster orchestration or scheduling of many heterogeneous jobs.
• Compiler, inference or serving work on large models.
• Simulation or benchmark environments (robotics sim, agent gyms, browser or operating-system environments, multi-agent benchmarks).
• Master's degree from a strong computer science program.
• Interest in joining an early-stage company for the long-term upside.

POSITION DETAILS

Work arrangement: Hybrid, three to four days a week in the San Francisco office
Schedule: Full-Time, six-day work week
Employment type: Permanent employment
Seniority: Mid-Level (1 to 5 years)
Start date: ASAP
Location: San Francisco, California, United States

Competitive base salary, founding-level equity, a performance bonus tied to customer deployments, and health, dental and vision coverage. Flexible arrangements for candidates outside the Bay Area are reviewed case by case.

WORK AUTHORIZATION

Candidates must be legally authorized to work in the United States. Visa sponsorship for new petitions is not available; H-1B transfers and OPT with sufficient time remaining are considered.