Staff ML Systems Research Engineer

reflectionai; San Francisco, New York (CA, NY); On-site; USD 180,000 - 280,000; Full time

Benefits offered by this job

Top-tier compensation

Stock options

Health & wellness

Meals in office

Parental leave

Unlimited vacation

Visa sponsorship

Team building

Job summary

Reflection seeks an exceptional ML researcher to design and build scalable language models, data pipelines, and training infrastructure. You will lead small projects, run rigorous experiments, and implement cutting-edge methods while collaborating with a fast-moving team.

Ideal candidates hold an MS or PhD in CS/ML, strong Python/PyTorch skills, and a passion for advancing the frontier of intelligence in a startup setting.

Qualifications

• Graduate degree in CS/ML or related field.
• Strong software engineering for large-scale systems.
• Experience with large-scale ETL workflows and data prep.
• Deep understanding of large-scale ML and language models.
• Proficient in Python and PyTorch.
• Balance research goals with engineering realities.
• Thrive in a fast-paced startup with action bias.
• Strong communication and collaboration.

Responsibilities

• Research and build algorithms, scaling laws, data processing, optimizers, and model architecture.
• Design experiments to advance scaling of large language models and data efficiency.
• Implement state-of-the-art methods from the deep learning literature.
• Lead small research projects independently while collaborating on larger initiatives.
• Optimize training infrastructure for efficient scaling.
• Contribute across the stack, from low-level optimizations to high-level model design.

Skills

Python

PyTorch

Distributed training

Large-scale ML

Data preprocessing

Software engineering

Communication

Education

MS or PhD in Computer Science or ML

Job description

Reflection seeks an exceptional ML researcher to design and build scalable language models, data pipelines, and training infrastructure. You will lead small projects, run rigorous experiments, and implement cutting-edge methods while collaborating with a fast-moving team.

Ideal candidates hold an MS or PhD in CS/ML, strong Python/PyTorch skills, and a passion for advancing the frontier of intelligence in a startup setting.