Forward Deployed Engineer (Training)

Baseten; New York (NY); On-site; USD 200,000 - 400,000; Full time

Benefits offered by this job

Medical, dental & vision for employee+

Flexible PTO including Winter Break

Paid parental leave

Fertility and family-building stipend

401(k) plan

Exposure to ML startups

Job summary

Baseten is seeking Forward Deployed Engineers who will work directly with leading AI customers to own technical outcomes on Baseten. You will manage workloads end-to-end, from problem framing and PoC development to production delivery, tackling complex inference and post-training challenges.

You will design evaluations, respond to incidents, and build automation to accelerate client engagements, while shaping the product roadmap with insights from earned experience across multiple accounts.

Qualifications

• 1-2 years of software engineering experience shipping and maintaining production code.
• Experience debugging complex production issues using logs, metrics, and traces.
• Comfort owning ambiguous technical problems and deciding when to involve others.
• Interest in working directly with customers and influencing the product.
• Clear communication on complex topics with customers or leadership.
• Genuine curiosity about AI inference and training and the infrastructure powering it.
• Willingness to participate in on-call rotations.

Responsibilities

• Act as the account's CTO on Baseten, accountable for workloads' design, run, and scale.
• Translate customer objectives from problem to PoC to production quickly using appropriate tools.
• Design evals and benchmarks to identify quality or performance gaps and close them.
• Respond to mission-critical failures with triage and ownership until fixes ship.
• Build internal tooling and automation for faster engagements and deployable references.
• Shape Baseten's roadmap by channeling customer needs into the product.
• Coordinate across multiple accounts, aligning stakeholders and managing risk.

Skills

Software engineering

Debugging production issues

Customer collaboration

Communication

Ambiguity tolerance

On-call experience

AI infrastructure interest

Education

Bachelor's degree in Computer Science

Tools

Kubernetes

Slurm

Ray

TensorRT

vLLM

PyTorch/JAX

Job description

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, led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to ship AI products.

THE ROLE

Forward Deployed Engineers work directly with the largest and fastest-growing AI companies in the world, owning their technical outcomes on Baseten and taking on the hardest problems in serving and improving models at scale. The work spans the model lifecycle: inference, post-training, and the systems that tighten the loop between them.

• Act as each account's de facto CTO on Baseten, with final accountability for how their workloads are designed, run, and scaled.
• Take customer objectives from vague to shipped: frame the problem, define the spec and success criteria, build the PoC, and carry it through to production quickly, using the right tools for the problem.
• Design the evals and benchmarks that isolate where quality or performance falls short, then close the gap yourself, whether that means optimizing inference, improving the model through post-training, or reworking the eval itself.
• Be the first responder to mission-critical failures including triage, owning the fix directly or route to the owning team and stay accountable until it ships.
• Build internal systems so that each engagement is faster than the last. This includes tooling and automation for eval and deployment infrastructure, and the recipes and reference implementations that make the product more self-serve.
• Shape the product itself, channeling what your accounts need into the roadmap and shipping fixes and features into Baseten's codebase yourself.
• Do all of this across multiple accounts at once, sequencing the work, pulling in the right people at the right time, and keeping customers and internal stakeholders aligned on status and risk.

Requirements

• Minimum 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 - working through logs, metrics, and traces to root-cause problems in unfamiliar systems
• Confidence owning ambiguous technical problems. This includes triaging, making decisions under uncertainty, and knowing when to pull in other engineers who own the underlying systems
• Motivation beyond pure engineering. An interest in wanting to work directly with customers, understand their problems firsthand, and influence the product
• Clear communication on complex technical topics, whether you're talking to a customer's engineers or their leadership
• Genuine curiosity about AI inference and training, and a drive 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
• Excitement about solving problems for some of the largest and fastest-growing companies in the world running mission-critical AI workloads

What You'll Bring

We don't expect any one person to cover all of this - the strongest candidates may spike in one or two of the following:

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

Benefits

• 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
• (U.S. only) Company-facilitated 401(k)
• Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

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).

Compensation Range: $200K - $400K