Research Engineer

Overview

Responsibilities

• Design and build our post-training infrastructure for SFT, RL and evaluation workflows.
• Build reliable pipelines for data preparation, dataset versioning, sampling, training, checkpoint management, and evaluation.
• Develop experiment orchestration and tracking systems that make runs reproducible, comparable, and easy to debug.
• Create reusable abstractions that allow researchers to launch experiments quickly across datasets, models, and training recipes.
• Integrate our systems with partner-lab training stacks, model APIs, compute environments, and evaluation infrastructure.

Required Qualifications

• At least 2 years of professional experience in machine learning engineering, research engineering, ML infrastructure, or a closely related field; 2-4+ years preferred.
• Strong Python and software-engineering skills.
• Hands on experience with PyTorch, JAX, Ray, and Slurm.
• Experience building production-quality ML training, evaluation, or data infrastructure.
• HAnds-on experience with LLM fine-tuning, post-training, and evaluation.
• Ability to build reliable, reproducible systems for launching and comparing ML experiments.
• Strong debugging skills across distributed systems, data pipelines, training infrastructure, and model behavior.
• Understanding of experimental design and the ability to extract actionable conclusions from noisy results.
• Ability to move quickly between infrastructure engineering and hands-on experimentation.
• A bias toward building, testing, and shipping.

Company Benefits (For Eligible Employees):

• Health Insurance Medical, Vision, Dental
• 401(k) With Employer Match
• Daily Meals Daily UberEats Stipend
• Wellness Stipend Monthly - Covers Equinox Membership
• Commute Covered
• We are an equal opportunity employer committed to providing a workplace free from discrimination and harassment. Employment decisions are made without regard to legally protected characteristics under applicable federal, state, or local law. We comply with applicable pay transparency requirements and provide compensation ranges based on the position, qualifications, experience, and other relevant factors. Reasonable accommodations are available to qualified individuals with disabilities and for sincerely held religious beliefs, as required by law. This job description is intended to describe the general nature and level of work performed and is not an exhaustive list of all duties, responsibilities, qualifications, or working conditions associated with the position. We reserve the right to modify this job description as business needs change.