Software Engineer - Model API's jobs in United States
cer-icon
Apply on Employer Site
company-logo

Baseten · 1 hour ago

Software Engineer - Model API's

positionUnited States
timeFull-time
remoteRemote
seniorityMid Level
date3+ years exp

Baseten is a company that powers mission-critical inference for leading AI companies. The Software Engineer - Model API's role involves designing and building the infrastructure for hosted API endpoints while ensuring performance and reliability of AI models in production.

Artificial Intelligence (AI)Developer ToolsMachine LearningSoftwareSoftware Engineering
check
H1B Sponsor Likelynote

Responsibilities

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

Qualification

Distributed systemsLarge-scale APIsPerformance profilingTensorRTCUDAAPI versioningDebugging complex systemsInfra instinctsWritten communicationCollaboration

Required

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

Preferred

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

Benefits

Competitive compensation, including meaningful equity.
100% coverage of medical, dental, and vision insurance for employee and dependents
Generous PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
Paid parental leave
Company-facilitated 401(k)
Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Company

Baseten

twittertwittertwitter
company-logo
Baseten is an AI infrastructure company that integrates machine learning into business operations, production, and processes.
dateFounded in 2019
location
San Francisco, California, USA
people-size51-200 employees
web-link
https://www.baseten.co

H1B Sponsorship

Baseten has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (6)
2024 (8)
2023 (1)
2020 (1)

Funding

Current Stage
Late Stage
Total Funding
$285M
Key Investors
BondGreylock
2025-09-05Series D· $150M
2025-02-19Series C· $75M
2024-03-04Series B· $40M

Leadership Team

leader-logo
Aaron Relph
Design
linkedin

Recent News

PYMNTS.com
Enterprise AI Shifts Focus to Inference as Production Deployments Scale
2025-12-13
Tech Startups - Tech News, Tech Trends & Startup Funding
Baseten acquires Parsed to double down on specialized AI over general-purpose models
2025-12-11
Tech Startups - Tech News, Tech Trends & Startup Funding
Skydo raises $10M Series A to fix cross-border payments as India’s exports target $2 trillion
2025-12-11
Company data provided by crunchbase