Engineering Manager - Inference Performance

Baseten

United States

Remote

USD 190,000 - 260,000

Full time

2 days ago
Be an early applicant
Application generator

Get a reply from this employer — a resume and cover letter tailored to exactly what they’re hiring for.

Get past ATS filters

Job summary

Baseten is seeking an Engineering Manager to lead part of the Inference Performance team, driving faster, more efficient AI workloads on GPUs. You will manage and grow a team across inference engine and runtime, setting direction, hiring, and supporting the people doing the work.

You’ll shape how the team is structured as it scales, balance customer needs with platform investments and help bring new model architectures to production quickly.

Qualifications

  • Bachelor's, Master's, or Ph.D. in CS/Engineering/Math required or related field.
  • Experience leading and growing a high-performance engineering team.

Responsibilities

  • Lead, mentor and grow a team of inference performance engineers through regular 1:1s and reviews.
  • Hire GPU and inference talent and foster a strong collaborative team culture as it scales.
  • Own the technical roadmap for runtime performance, balancing customer needs and model launches.
  • Review designs, guide profiling and optimization, and reason from first principles about time and memory.

Skills

Team leadership
Mentoring
Hiring top talent
Performance optimization
GPU performance

Education

Bachelor's, Master's, or Ph.D. in Computer Science, Engineering, Mathematics

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

We're looking for an Engineering Manager to lead part of our Inference Performance team. This team makes the world's most demanding AI workloads run faster and more efficiently on GPUs. You'll manage and grow a team of inference performance engineers working across the inference engine and runtime: kernels, scheduling, batching, KV-cache management, speculative decoding and prefill/decode disaggregation. This is a hands-on technical leadership role. You'll set direction, unblock hard problems and earn the team's trust by going deep on GPU performance, while also hiring, developing and supporting the people doing the work. Your team's output directly affects how fast our customers' models run and how efficiently we serve them. The team is scaling quickly, so you'll help shape how it is structured as it grows.

EXAMPLE INITIATIVES

Your team will work on these types of projects as part of our Inference Runtime team:

  • Agentic inference optimization: 50-90% faster engines
  • Agentic Kernels in Production
  • Live draft model training for speculative decoding
  • The Baseten Inference Stack
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.
REQUIREMENTS
  • Bachelor's, Master's, or Ph.D. degree in Computer Science, Engineering, Mathematics, or a related field.
  • Experience managin
Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Software Engineer- Inference Performance
Software Engineer- Inference Performance

Baseten • United States

Remote
USD 150,000 - 210,000
Engineering Manager - Inference Performance
Engineering Manager - Inference Performance

Candidate • San Francisco (CA)

On-site
USD 220,000 - 270,000
Equity
US medical/dental/vision
Flexible PTO
+3
Software Engineer- Inference Platform
Software Engineer- Inference Platform

Baseten • United States

Remote
USD 180,000 - 240,000
Software Engineer - GPU Kernels
Software Engineer - GPU Kernels

Baseten • United States

Remote
USD 140,000 - 210,000
Hands-on Engineering Manager, GPU Inference
Hands-on Engineering Manager, GPU Inference

Baseten • United States

Remote
USD 190,000 - 260,000
Software Engineer- Inference Performance
Software Engineer- Inference Performance

Baseten • San Francisco (CA), New York (NY), Seattle (WA)

On-site
USD 180,000 - 260,000
Meaningful equity
Medical, dental, vision for employee +
Flexible PTO with Winter Break
+4
Software Engineer- Model Performance Systems
Software Engineer- Model Performance Systems

Baseten • San Francisco (CA)

On-site
USD 160,000 - 200,000
Competitive compensation
Equity
Medical/dental/vision insurance
+4
Software Engineer- Inference Performance Baseten · New York City, NY Full-time · Hybrid $180,000–360,000 25 minutes ago
Software Engineer- Inference Performance Baseten · New York City, NY Full-time · Hybrid $180,000–360,000 25 minutes ago

Emploive • New York (NY)

On-site
USD 180,000 - 360,000
Equity
Insurance for dependents
Winter Break
Technical Program Manager, Model Performance
Technical Program Manager, Model Performance

Baseten • New York (NY)

On-site
USD 165,000 - 330,000
Equity
Medical, dental, vision insurance
Flexible PTO
+3
Technical Program Manager, Model Performance
Technical Program Manager, Model Performance

Baseten • San Francisco (CA), Northern (KY)

On-site
USD 170,000 - 230,000
Competitive compensation
Equity
Medical/dental/vision insurance
+4