Staff ML Research Engineer - Open Models

Reflection; San Francisco (CA); On-site; USD 180,000 - 260,000; Full time

Benefits offered by this job

Salary and equity

Stock options

Health & wellness

Meals provided

Parental leave 22 weeks

Unlimited PTO (US) / 30 days UK

Visa sponsorship

Team off-sites

Job summary

Reflection is a research lab building open models that let anyone control their intelligence and help shape the future of AI. We source, curate, and scale data responsibly to fuel cutting-edge models in a collaborative environment.

You will contribute to data collection pipelines, quality metrics, and pre-training experiments, partnering with infrastructure and safety teams. Join us in San Francisco to advance open AI while growing your research career.

Qualifications

• Bachelor’s degree or equivalent in CS/ML/Physics/Math or related discipline.
• Strong written communication and the ability to explain complex technical ideas.
• Proficiency in Python and at least one DL framework; debugging distributed workloads.

Responsibilities

• Design and implement data sourcing, curation, and filtering pipelines for large-scale data.
• Develop data quality metrics measuring coverage, diversity, and representativeness.
• Join pre-training research to design and run data experiments with scalable pipelines.
• Investigate data risks including privacy, safety, and licensing concerns.
• Evaluate dataset changes through ablations and scaling experiments.
• Publish research and share code, datasets, and insights with the open AI community.

Skills

Python

PyTorch/JAX

Distributed workloads debugging

Education

Bachelor’s degree in CS/ML/Physics/Math or related field

PhD in CS/ML/Physics/Math or equivalent industry experience

Job description

Reflection is a research lab building open models that let anyone control their intelligence and help shape the future of AI. We source, curate, and scale data responsibly to fuel cutting-edge models in a collaborative environment.

You will contribute to data collection pipelines, quality metrics, and pre-training experiments, partnering with infrastructure and safety teams. Join us in San Francisco to advance open AI while growing your research career.