Head of GPU Inference Performance

US Health Partners, LLC; New York, Northern (NY, KY); Hybrid; USD 170,000 - 250,000; Full time

Benefits offered by this job

Competitive compensation

Equity

Medical/Dental/Vision insurance

Flexible PTO with Winter Break

Paid parental leave

Fertility and family-building stipend

401(k)

Learning opportunities

Job summary

Baseten is seeking an Engineering Manager to lead part of the Inference Performance team. You will mentor a group of inference performance engineers and drive GPU-focused optimization across kernels, scheduling, and KV-cache management. The role blends technical depth with people leadership to accelerate model workloads on GPUs.

You will hire, develop and support engineers, set direction, review designs, and partner across teams to deliver fast, efficient inference at scale for customers.

Qualifications

• Bachelor's, Master’s, or Ph.D. in Computer Science, Engineering, Mathematics, or a related field.
• Experience managing engineers, including hiring, mentoring, giving feedback and running performance reviews.
• Experience leading or closely supporting GPU optimization teams in training, inference or recommendation systems.
• Strong technical depth in GPU workloads, with a solid understanding of GPU architecture and performance tradeoffs.
• Familiarity with ML libraries such as PyTorch, TensorRT or TensorRT-LLM.
• A track record of driving roadmaps and shipping complex technical projects with a team.
• Clear written and verbal communication, including the ability to align stakeholders across teams.

Responsibilities

• Lead, mentor and grow a team of inference performance engineers through regular 1:1s, clear feedback, career development and performance reviews.
• Hire top GPU and inference engineering talent, and build a strong, collaborative team culture as the runtime team scales.
• Own the technical roadmap and execution for runtime performance work, balancing customer needs, new model launches and long-term platform investments.
• Stay close to the technical work. Review designs, guide profiling and optimization efforts, and help the team reason from first principles about where time and memory go.
• Drive the productionization of inference techniques such as quantization, speculative decoding, KV-cache reuse, chunked prefill and custom scheduling.
• Turn performance wins into measurable outcomes: tokens per GPU-hour, utilization, latency and cost.
• Help the team bring up and tune new model architectures on new hardware quickly, often in the same week they're released.
• Partner with Infrastructure, Inference Platform, Kernels, Model APIs and customer-facing teams to set priorities, coordinate launches and ship wins.
• Set high standards for engineering quality, benchmarking, operational excellence and incident response.

Skills

GPU performance

Team leadership

Hiring

Profiling

CUDA

Education

Bachelor's/Master's/Ph.D. in CS/Engineering

Tools

PyTorch

TensorRT

TensorRT-LLM

vLLM

CUDA

Triton

CUTLASS

Job description

Baseten is seeking an Engineering Manager to lead part of the Inference Performance team. You will mentor a group of inference performance engineers and drive GPU-focused optimization across kernels, scheduling, and KV-cache management. The role blends technical depth with people leadership to accelerate model workloads on GPUs.

You will hire, develop and support engineers, set direction, review designs, and partner across teams to deliver fast, efficient inference at scale for customers.