Research Scientist

About The Role

We're looking for a Research Scientist to investigate how intervening on training data can improve the quality and shape the behavior of deep learning models. You'll source and implement ideas from the literature, conduct research grounded in real customer needs, and collaborate closely with engineers and product teams to turn findings into tangible impact. This role requires strong scientific judgment, fluency with the deep learning literature, and the drive to work autonomously in a fast-moving startup environment.

About The Company

At DatologyAI, we’ve built a state of the art data curation suite to automatically curate and optimize petabytes of data to create the best possible training data for your models. Training on curated data can dramatically reduce training time and cost (7-40x faster training depending on the use case), dramatically increase model performance as if you had trained on >10x more raw data without increasing the cost of training, and allow smaller models with fewer than half the parameters to outperform larger models despite using far less compute at inference time, substantially reducing the cost of deployment. For more details, check out our recent research on synthetic data scaling (BeyondWeb) and pretraining with domain-specific data (The Finetuner’s Fallacy).

What You'll Work On

• Sourcing, vetting, implementing, and improving promising ideas from the literature and your own thinking.
• Conducting research grounded in real customer needs and product outcomes.
• Collaborating closely with engineers, talking to customers, and shaping the product vision.

About You

• 3+ years of deep learning research experience
• Strong fundamentals in deep learning
• Praactical experience and/or publications in one or more of the following areas:
  • Data pruning and curation
  • Curriculum learning
  • Synthetic data generation
  • Dataset distillation
  • Effects of training data on model behavior
  • Embedding models and semantic search
  • Training large vision (including video), language, or multimodal models
  • Efficient MLE
• Enough software engineering and PyTorch experience (or willingness to learn) to run large-scale experiments and build production prototypes
• A demonstrated track record in deep learning research, whether through papers, tools, or other artifacts

Nice To Have

• Experience with distributed data processing tools like Spark or Snowflake
• Experience building and shipping ML products

Compensation

• The salary for this position ranges from $180,000 to $300,000.
• Starting pay is based on job-related skills, experience, qualifications, and interview performance.
• Our benefits are built to support your well-being and growth:
  • 100% covered health benefits (medical, vision, and dental)
  • 401(k) plan with a generous 4% company match
  • Unlimited PTO policy
  • Paid Parental Leave of 12 weeks, plus 6 months of WFH flexibility
  • 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.