Member of Technical Staff - Pre-Training Data
San Francisco, CA • FullTime
Posted 5h ago
Job Location
San Francisco, CA
Tech Stack
Remote Work Policy
On-site
Employment Type
FullTime
Categories
AI Research Engineer
About the job
Reflection is a research lab dedicated to making intelligence open and accessible for everyone. We build open models that empower individuals to control their intelligence and shape the future of AI. This role focuses on the foundational aspects of AI development, specifically in pre-training data, contributing to the creation of open foundational models. You will be part of a talented team, helping to define the company's future and the future of open AI.
Responsibilities
- Design and implement techniques for sourcing, curating, and filtering large-scale web, text, and code data.
- Develop data quality metrics and analyses to measure coverage, diversity, and representativeness.
- Design and run data experiments for pre-training research, collaborating with infrastructure teams to scale data systems.
- Investigate and mitigate data risks, including privacy, safety, and licensing concerns.
- Evaluate dataset changes through experiments and trace their downstream effects on model learning and behavior.
- Publish and present research, and share code, datasets, and insights.
Requirements
- Proficiency in Python and at least one deep learning framework (e.g., PyTorch or JAX).
- Experience debugging distributed workloads and writing scalable code.
- Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline.
- Strong theoretical and empirical grounding.
- Clear written communication and ability to explain complex technical ideas.
Benefits
- Top-tier compensation: Salary and equity structured to recognize and retain talent globally.
- Stock options.
- Comprehensive medical, dental, vision, and life insurance.
- Annual wellness allowance.
- Lunch and dinner provided in the office daily.
- 22 weeks paid parental leave for all new parents.
- Unlimited paid time off in the U.S. and 30 days in the U.K.
- Sponsorship support for visas and long-term immigration pathways.
- Regular off-sites, happy hours, and team celebrations.