Forward Deployed Engineer (Training)

ABOUT BASETEN

Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma and Writer. 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. 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 to ship AI products.

THE ROLE

As a Forward Deployed Engineer at Baseten, you will partner directly with customers to architect, build, and deploy high-scale post-training solutions on Baseten’s platform. You’ll own the journey with customers from initial exploration to production deployment, translating ambiguous business goals into performant fine-tuning pipelines.

This role is a great fit for entrepreneurial engineers who want a front-row view into how modern companies adopt post-training at scale. You’ll work across product, software development, research, and customer teams to deliver performant and observable post-training pipelines.

To be clear, this is a technical, hands-on role focused on software development, architecture design, and experiment design, with aspects of product management, technical customer success, and pre-sales solution engineering mixed in.

THE PRODUCT

Take a look at what we’ve built so far:

• Baseten Loops beta launch https://www.baseten.co/blog/introducing-the-baseten-loops-sdk/
• Training Infra GA launch https://www.baseten.co/blog/baseten-training-is-ga/#training-is-now-ga
• Autoresearch on Baseten https://www.baseten.co/blog/baseten-training-an-autoresearch-substrate/

RECENT RESEARCH

• 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

• Develop and maintain software systems and product features using Python in a production-level environment.
• Drive customer impact by designing, implementing, and deploying Baseten solutions end-to-end: problem framing, evaluations, post-training approach and experiment design, experimentation, quantization, and production deployment. This involves working with customers’ engineering teams at every stage of the customer journey, including sales, implementation, and expansion.
• Deliver with velocity: turn vague objectives into clear specs and well-defined PoCs so we can rapidly ship well-tested services and outcomes for our customers.
• Optimize and enhance our post-training solutions, contributing to the continuous improvement of our technical stack. This includes developing features and PRDs with engineering and product teams.
• Own products and customer projects end-to-end, functioning as an engineer, project manager, and product manager, with a focus on user empathy, project specification, and end-to-end execution.
• Navigate ambiguity and exercise good judgment on the tradeoffs and tools needed to solve problems, avoiding unnecessary complexity.
• Demonstrate pride, ownership, and accountability for your work, and expect the same from your teammates.

REQUIREMENTS

• Experience working with RL and SFT for large language models.
• Bachelor's, Master's, or Ph.D. degree in Computer Science, Engineering, Mathematics, or a related field.
• 2+ years of professional work experience in a fast-paced, high-growth environment.
• Python proficiency.
• Strong communication skills, particularly on complex technical topics.

BENEFITS

• Competitive compensation, including meaningful equity.
• 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
• Company-facilitated 401(k)
• Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Apply now to embark on a rewarding journey in shaping the future of AI! 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.

At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.

We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law (by example, the requirements of the San Francisco Fair Chance Ordinance, where applicable).