Research Scientist, Post-Training

About the Company

Companies want to train their own large models on their own data. The current industry standard is to train on a random sample of your data, which is inefficient at best and actively harmful to model quality at worst. There is compelling research showing that smarter data selection can train better models faster—we know because we did much of this research. Given the high costs of training, this presents a huge market opportunity. We founded DatologyAI to translate this research into tools that enable enterprise customers to identify the right data on which to train, resulting in better models for cheaper. Our team has pioneered deep learning data research, built startups, and created tools for enterprise ML. For more details, check out our recent blog posts sharing our high-level results for text models and image-text models.

We've raised over $57M in funding from top investors like Radical Ventures, Amplify Partners, Felicis, Microsoft, Amazon, and notable angels like Jeff Dean, Geoff Hinton, Yann LeCun and Elad Gil. We're rapidly scaling our team and computing resources to revolutionize data curation across modalities.

This role is based in Redwood City, CA. We are in office 4 days a week.

About the Role

We’re looking for a Research Scientist to lead work on post-training data curation for foundation models. You’ll design and implement algorithms to generate and improve instruction, preference, and other post-training datasets. You’ll also help bridge the gap between pre-training and post-training by exploring how to jointly optimize data across stages. This role requires strong scientific judgment, fluency with the deep learning literature, and a drive to turn research ideas into real-world impact. You’ll work autonomously, collaborate closely with engineers and product teams, and shape the future of data curation at DatologyAI.

What You'll Work On

• Post-training data curation. You’ll conduct research on how to algorithmically curate post-training data—e.g., how to generate and refine preference and instruction-following data, how to curate capability- and domain-specific data, and make post-training more effective, controllable, and generalizable.

• Unifying pre-training and post-training data curation. Pushing the bounds on model capabilities requires unifying post-training and pre-training data curation. You will pursue research on end-to-end data curation: how to curate pre-training data to improve the post-trainability of models and how to jointly optimize pre- and post-training data curation, all in service of maximizing the final performance of post-trained models.

• Transform messy literature into practical improvements. The research literature is vast, rife with ambiguity, and constantly evolving. You will use your skills as a scientist to source, vet, implement, and improve promising ideas from the literature and of your own creation.

• Conduct science driven by real-world needs. At DatologyAI, we understand that conference reviewers and academic benchmarks don’t always incentivize the most impactful research. Your research will be guided by concrete customer needs and product improvements.

How You'll Work

• Nobody knows how to do your work better than you. We believe that scientists do their best work when they have the autonomy to pursue problems in the manner they prefer, and we will ensure that you are equipped with the context and resources you need to succeed.

• Science is more than just experiments. We expect our Research Scientists to collaborate closely with engineers, talk to customers, and shape the product vision.

About You

• 3+ years of deep learning research experience

• Experience with post-training large vision, language, and multimodal models

• Post-training algorithm development, data curation, and/or synthetic data methods for:

  • Preference-based tuning (e.g. DPO, RLVR, RRHF)

  • Alternative supervision & self-supervision techniques such as self-training and chain-of-thought distillation

  • SFT (e.g. instruction tuning and demonstration fine-tuning)

• Post-training tooling development and engineering experience

• Strong understanding of the fundamentals of deep learning

• Sufficient software engineering + deep learning framework (PyTorch or a willingness to learn PyTorch) skills to conduct large-scale research experiments and build production prototypes.

• Demonstrated track record of success in deep learning research, whether papers, tools, or other research artifacts.

We would love it if candidates have:

• Experience with data management and distributed data processing solutions (e.g. Spark, Snowflake, etc.)

• Experience building + shipping ML products

Candidates do not need a PhD or extensive publications. Some of the best researchers we’ve worked with have no formal training in machine learning, and obtained all of their experience by working in industry and building products. We believe that adaptability, combined with exceptional communication and collaboration skills are the most important ingredients for successful research in a startup environment.

Compensation

At DatologyAI, we are dedicated to rewarding talent with highly competitive salary and significant equity. The salary for this position ranges from $180,000 to $260,000.

• The candidate's starting pay will be determined based on job-related skills, experience, qualifications, and interview performance.

Benefits

We offer a comprehensive benefits package to support our employees' well-being and professional growth:

• 100% covered health benefits (medical, vision, and dental).

• 401(k) plan with a generous 4% company match.

• Unlimited paid time off (PTO) policy.

• Annual $2,000 wellness stipend.

• Annual $1,000 learning and development stipend.

• Daily lunches and snacks are provided in our office!

• Relocation assistance for employees moving to the Bay Area.