Baseten
Post-Training Applied Researcher
San Francisco
Sponsorship not specifiedDetected 9 days ago
Machine LearningSparkLLMsMLOpsResearch
Stay score
odds of building a lasting career here
40Risky
Cap-exempt (no lottery)0
Sponsors this role90
Entry-level history0
PERM / green-card track0
Lottery odds40
Fits your clock70
Thin sponsorship signal and lottery-bound. A low-probability bet with your clock running. Prioritize cap-exempt roles and proven entry-level sponsors first.
Lottery odds assume a STEM candidate.
Personalize to your clock →Community outcomes
No reports yet — be the first to help the next applicant.
About the role
- Dense, on-policy or both? https://www.baseten.co/research/dense-on-policy-or-both/
- Repeated kv cache for long-running agents https://www.baseten.co/research/repeated-kv-cache-for-long-running-agents/
- Distillation without the dark - replicating black-box on-policy distillation on Baseten https://www.baseten.co/research/distillation-without-the-dark/
Responsibilities
- Design and run post-training pipelines: SFT, GRPO, DPO, RLVR, reward function engineering, and synthetic data generation.
- We're growing quickly and recently raised our $1.5B Series F https://www.baseten.co/blog/announcing-our-series-f/, led by Altimeter Capital, Conviction Partners, and Spark Capital.
- Join us and help build the platform engineers turn to ship AI products.
Requirements
- Strong intuition for reward engineering: the ability to distinguish between a reward that trains effectively and one that will exploit at scale
- Comfort working across the full pipeline from dataset construction through training, evaluation, and deployment
- Experience with production ML systems.
- Hands-on experience training LLMs with reinforcement learning - demonstrated understanding of GRPO or PPO beyond recipe-level reproduction, including group advantage computation, clipped objectives, and KL penalty design
- Experience building multi-turn agent environments with tool use, not limited to single-turn question-answering setups
- Experience with production ML systems. Preference for candidates who have closed a training-inference loop where production data feeds back into model improvement
- Competitive compensation, including meaningful equity
- (U.S. only) 100% coverage of medical, dental, and vision insurance for employee and dependents
- Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
- Paid parental leave
- Fertility and family-building stipend through Carrot
Nice to have
- Experience with RL training frameworks
- Publications at NeurIPS, ICML, ICLR, focused on RL for LLMs, reward modeling, or alignment
Compensation
- Competitive compensation, including meaningful equity
Benefits
- Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.
- If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.
- Build task-specific training environments and evals tailored to customer domains like healthcare, code generation, and legal, spanning multi-turn tool use, sandboxed execution, and agentic workflows.
- By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production.
Company info
- We are looking for people with hands-on LLM fine-tuning and RL experience.
- At Baseten, we are committed to fostering a diverse and inclusive workplace.
- Work directly with customers to translate production data into training signal, designing reward loops from real usage patterns and handling distribution shift.
Apply directly at Baseten →Create a free account for alerts like thisView Baseten immigration profile
This listing is sourced directly from Baseten's careers page and normalized into a canonical job model.