Software Engineer - Model Products

🏢 ОфисMiddleПолная занятость

Зарплата не указана

Вакансия висит подозрительно долго, вилки нет, про деньги придётся договариваться с нуля.

Нажмите на сигнал, чтобы увидеть, на чём он основан

Наша компания

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 to ship AI products.

О роли

Baseten’s Model Performance (MP) team is responsible for ensuring the models running on our platform are fast, reliable, and cost‑efficient. As part of this team, you’ll focus on Model API's — the infrastructure powering our hosted API endpoints for the latest open‑source models. This work spans distributed systems, model serving, and developer experience. You’ll join a small, high‑impact team operating at the intersection of product, model performance, and infra, helping to define how developers interact with AI models at scale

Чем предстоит заниматься

Design, build, and operate the Model APIs surface with focus on advanced inference capabilities: structured outputs (JSON mode, grammar-constrained generation), tool/function calling and multi-modal serving

Profile and optimize TensorRT-LLM kernels, analyze CUDA kernel performance, implement custom CUDA operators, tune memory allocation patterns for maximum throughput and optimize communication patterns across multi-GPU setups

Productionize performance improvements across runtimes with deep understanding of their internals: speculative decoding implementations, guided generation for structured outputs, custom scheduling and routing algorithms for high-performance serving

Build comprehensive benchmarking frameworks that measure real-world performance across different model architectures, batch sizes, sequence lengths, and hardware configurations

Productionize performance improvements across runtimes (e.g.TensorRT, TensorRT‑LLM): speculative decoding, quantization, batching, and KV‑cache reuse

Instrument deep observability (metrics, traces, logs) and build repeatable benchmarks to measure speed, reliability, and quality

Implement platform fundamentals: API versioning, validation, usage metering, quotas, and authentication

Collaborate closely with other teams to deliver robust, developer‑friendly model serving experiences

Наши требования

3+ years experience building and operating distributed systems or large‑scale APIs

Proven track record of owning low‑latency, reliable backend services (rate‑limiting, auth, quotas, metering, migrations)

Infra instincts with performance sensibilities: profiling, tracing, capacity planning, and SLO management

Comfortable debugging complex systems, from runtime internals to GPU execution traces

Strong written communication; able to produce clear design docs and collaborate across functions

Experience with LLM runtimes (vLLM, SGLang, TensorRT‑LLM) or contributions to open-source inference engines (vLLM, TensorRT-LLM, SGLang, TGI)

Knowledge of Kubernetes, service meshes, API gateways, or distributed scheduling

Background in developer‑facing infrastructure or open‑source APIs

We value infra‑leaning generalists who bring strong engineering fundamentals and curiosity. ML experience is a plus, but not required

Мы предлагаем

Competitive compensation, including meaningful equity

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

Company-facilitated 401(k)

Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities

Baseten

San Francisco

ГрейдMiddle

ЗанятостьПолная занятость

РегионСША

ФорматОфис

ИсточникСкрыто

Опубликовано11 окт.

Все вакансии компании

AI-помощник

под эту вакансию

Войди, чтобы AI оценил твоё соответствие вакансии и написал сопроводительное письмо

Мы против мошенников на площадке: если тебя просят заплатить, продиктовать код или установить непонятное приложение, прекращай общение и сразу пиши нам (чат с основателем или форма обратной связи).