B

Baseten

Runs AI models for production software

Forward Deployed Engineer - Training

AI Application EngineeringConfirmed live in the last 24 hours
Full-timeHybrid

About the Job

Requirements

  • At least 1โ€“2 years of software engineering experience shipping and maintaining code in large production systems, ideally with breadth across the stack.
  • Experience debugging complex production issues by working through logs, metrics, and traces to root-cause problems in unfamiliar systems.
  • Confidence owning ambiguous technical problems, including triaging, making decisions under uncertainty, and knowing when to involve engineers who own underlying systems.
  • Interest in working directly with customers, understanding their problems firsthand, and influencing the product.
  • Ability to communicate complex technical topics clearly to customer engineers and leadership.
  • Curiosity about AI inference and training, and motivation to become an expert in the infrastructure powering it.
  • Willingness to respond to customers outside regular working hours and participate in an on-call rotation.

Responsibilities

  • Act as each account's de facto CTO on Baseten, with final accountability for how workloads are designed, run, and scaled.
  • Take customer objectives from problem framing through specification, success criteria, proof of concept, and production delivery.
  • Design evaluations and benchmarks to identify quality or performance gaps, then address them through inference optimization, post-training, or evaluation changes.
  • Respond to mission-critical failures by triaging, owning the fix or routing it to the responsible team, and remaining accountable until it ships.
  • Build tooling, automation, recipes, and reference implementations that improve evaluation and deployment infrastructure and make the product more self-serve.
  • Shape the product roadmap from account needs and ship fixes and features into Baseten's codebase.
  • Sequence work across multiple accounts, involve the right people, and keep customers and internal stakeholders aligned on status and risk.

Desired Qualifications

  • Depth in an infrastructure domain such as storage systems or networking, from cloud layers to cluster interconnects such as InfiniBand and RoCE.
  • Experience operating distributed compute platforms such as Kubernetes, Slurm, or Ray, especially for GPU workloads.
  • Detailed understanding of LLM architectures and modern inference engines such as vLLM, TensorRT-LLM, or SGLang.
  • Ability to profile and optimize GPU workloads in training or serving.
  • Hands-on experience with post-training techniques such as supervised fine-tuning (SFT) and reinforcement learning (RL), or a broader deep learning background plus fluency in a tensor computation library such as PyTorch or JAX.
  • Operational depth in running on-call, leading incident response, and debugging distributed systems under pressure.

About the Company

Baseten

Runs AI models for production software

Baseten runs AI models for software companies that need fast, reliable responses inside their products, from text generation to speech and image processing. Its Inference Stack combines optimized model runtimes with infrastructure that deploys, serves, and monitors custom or open models in production. Teams package models with its open source Truss tool, while Model APIs let developers test ready-to-use models and Training runs jobs that adapt models to their own data. Customers pay for computing time or tokens processed, with negotiated plans for larger deployments and support needs. Baseten was founded in 2019 and is headquartered in San Francisco.

Company size
201-500
Company stage
Series F
Total funding
$2.1B
Headquarters
San Francisco, CA
Founded
2019

Recently Posted Jobs

Explore more opportunities at Baseten

See more jobs