Skip to content
BasetenBasetenSan Francisco, CA

Post-Training Research Scientist

Conducts research on post-training methodologies and performant inference for AI models, balancing pure research with applied work for production systems. Requires PhD in ML with top publications and ability to design rigorous experiments at scale.

210k – 285k/yr
HybridAI Research

About the role

Recent Research

Responsibilities

  • Define and pursue a research agenda spanning both pure and applied work, with the applied component connected to Baseten's platform and customer needs
  • Design and execute rigorous experiments, frequently at meaningful scale (multi-node, 1T+ parameter models)
  • Publish at top venues (NeurIPS, ICML, ICLR) and establish Baseten's research presence
  • Collaborate with model performance and training infrastructure teams to bridge research findings and production systems
  • Mentor junior researchers and shape the technical direction of the research organization as it grows

Qualifications

  • PhD or equivalent research depth in machine learning, with first-author publications at top venues
  • Demonstrated ability to move from theory through implementation to empirical results — not exclusively theoretical or exclusively engineering work
  • Judgment about problem selection, the ability to distinguish research that advances a metric from research that changes how systems are built
  • Willingness to operate in a startup environment where the majority of research informs product decisions, with timelines measured in months rather than years

Preferred Qualifications

  • Experience with production ML systems and an understanding of the constraints that cause academic solutions to fail in deployment
  • Background spanning multiple research areas (e.g., both interpretability and RL, or both systems and training methodology)
  • Track record of open-source contributions or community building in ML research

Benefits

  • Competitive compensation, including meaningful equity.
  • 100% coverage of medical, dental, and vision insurance for employee and dependents
  • Generous PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
  • Paid parental leave
  • Company-facilitated 401(k)
  • Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Skills

Machine LearningPost-TrainingInference OptimizationReinforcement LearningInterpretabilityProduction Ml SystemsPyTorchJAXLLMsDistributed Training

Similar roles

AI Research jobs
Abridge

Machine Learning Scientist (All Levels)

AbridgeSan Francisco, CA +2

Conducts machine learning research in medical NLP for conversation summarization, evidence extraction, and outcome prediction. Publishes at top AI conferences, deploys models to production, and requires MS/PhD plus strong PyTorch/TensorFlow experience.

205k – 300k/yr
HybridAI Research
OpenAI

Research Engineer, Frontier Evals & Environments

OpenAISan Francisco, CA

Builds ambitious RL environments and evaluation systems to measure and steer frontier AI models toward safe AGI. Requires strong ML research engineering, statistical skills, and red-teaming mindset for end-to-end project ownership in fast-paced setting.

205k – 380k/yr
On-siteAI Research
Scale AI

Research Scientist, Safety Post Training

Scale AISan Francisco, CA +1

Develop and apply post-training methods and interpretability techniques to improve safety and understanding of frontier AI systems. Requires 3+ years of ML experience, expertise in RL techniques like RLHF and DPO, and published research in generative AI.

216k – 270k/yr
Hybrid3+ YOEAI Research
Postman

Applied AI Scientist, Small Language Model and AI Training

PostmanSan Francisco, CA

Leads R&D on small language models and AI training, developing efficient architectures, optimizing performance, and ensuring safety. Collaborates with research, engineering, and product teams using Python, PyTorch, TensorFlow, or JAX.

219k – 276k/yr
HybridAI Research
Hedra

Research Scientist

HedraSan Francisco, CA

Leads original research in action-conditioned world models, physical AI, and generative modeling for embodied systems. Requires PhD in ML/CS/Robotics with top publications and expertise in generative models and large-scale training.

200k – 325k/yr
On-siteAI Research