Forward-Deployed AI Engineer: Scale Inference & Ship Solutions

Triwill Group; San Francisco, Northern (CA, KY); Hybrid; USD 140,000 - 190,000; Full time

Benefits offered by this job

Equity

Medical, dental, vision coverage fore

Flexible PTO

Parental leave

Fertility stipend

401(k) plan

Exposure to ML startups

Job summary

Baseten powers mission-critical AI inference for leading companies, and is seeking Front Deployment Engineers to work directly with top AI customers, owning outcomes from PoC to production. You will design workloads, frame problems, and ship robust solutions, spanning inference, post-training, and deployment systems.

You will shape evals, triage incidents, and build scalable internal tooling, while collaborating across multiple accounts to keep stakeholders aligned and drive product improvements

Qualifications

• Minimum 1-2 years of software engineering experience in large production systems.
• Experience debugging complex production issues using logs, metrics, and traces to root-cause problems in unfamiliar systems.
• Ability to own ambiguous technical problems, including triaging, making decisions under uncertainty, and knowing when to pull in other engineers.
• Interest in working directly with customers, understanding their problems, and influencing the product.
• Clear communication on complex technical topics to customers or 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 hours and participate in an on-call rotation.
• Excitement about solving problems for some of the largest and fastest-growing companies running mission-critical AI workloads.

Responsibilities

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

Skills

Software engineering

Debugging production issues

Customer interaction

Communication

AI infrastructure

On-call experience

Problem ownership

Tools

Kubernetes

Slurm

Ray

TensorRT-LLM

vLLM

PyTorch

JAX

Job description

Baseten powers mission-critical AI inference for leading companies, and is seeking Front Deployment Engineers to work directly with top AI customers, owning outcomes from PoC to production. You will design workloads, frame problems, and ship robust solutions, spanning inference, post-training, and deployment systems.

You will shape evals, triage incidents, and build scalable internal tooling, while collaborating across multiple accounts to keep stakeholders aligned and drive product improvements