Forward Deployed Engineer (Training)

Baseten

$120K — $150K *
Information Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Experience with Reinforcement Learning (RL) and Supervised Fine-Tuning (SFT) for large language models.
  • Bachelor's, Master's, or Ph.D. in Computer Science, Engineering, Mathematics, or related field.
  • 2+ years of experience in a fast-paced, high-growth environment.
  • Proficiency in Python programming.
  • Strong communication skills for complex technical discussions.

Responsibilities

  • Develop and maintain high-performance software systems and product features using Python.
  • Design, implement, and deploy solutions end-to-end, working with customers throughout their journey.
  • Transform vague objectives into clear specifications and proofs of concept for rapid deployment.
  • Continuously enhance post-training solutions and technical stack by collaborating with engineering and product teams.
  • Take ownership of product and customer projects, acting as engineer, project manager, and product manager.
  • Effectively navigate ambiguity and demonstrate sound judgment in problem-solving.
  • Show pride and accountability in your work and foster the same in your teammates.

Benefits

  • Competitive compensation package including substantial equity.
  • Full medical, dental, and vision insurance coverage for employees and their dependents.
  • Flexible PTO policy with a company-wide Winter Break allowing time off from Christmas Eve to New Year's Day.
  • Paid parental leave for employees.
  • Fertility and family-building stipend provided through Carrot.
  • Company-supported 401(k) plan.
  • Exposure to various ML startups, enabling exceptional learning and networking opportunities.
Full Job Description
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
  • Training Infra GA launch
  • Autoresearch on Baseten

RECENT RESEARCH
  • Dense, on-policy, or both?
  • Repeated KV cache for long-running agents
  • Distillation without the dark - replicating black-box on-policy distillation on Baseten

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.

Similar Jobs

More Jobs at Baseten

More Information Technology Jobs

Find similar Forward Deployed Engineer (Training) jobs: