Jobs at Baseten

Manage and mature Baseten’s security governance, risk, compliance, privacy, audit, certification, vendor assessment, and customer assurance programs. Translate regulatory and customer requirements into technical controls and product requirements, partnering with Engineering, Product, Security, Legal, and Operations. Coordinate SOC 2, ISO, HIPAA, FedRAMP, and related audits; manage risk assessments, evidence collection, remediation, security questionnaires, training, and continuous improvement in a fast-growing AI platform company.

Develop and productionize high-performance LLM inference techniques across runtimes, scheduling, serving, routing, and GPU kernels. Profile and optimize latency, throughput, memory usage, and cost; support new models and hardware; build benchmarking frameworks; and contribute to open-source inference engines such as vLLM, SGLang, and TensorRT-LLM. The role requires strong programming, GPU architecture, ML library, and LLM optimization knowledge.

Build and operate the distributed inference platform powering large-scale LLM deployments. Responsibilities include orchestration, routing, autoscaling, scheduling, Model APIs, authentication, quotas, metering, observability, benchmarking, and production reliability across Kubernetes, networking, distributed runtimes, and GPU workloads. The role owns projects end to end and partners with performance engineers to deliver efficient, scalable AI inference capabilities and strong developer experiences.

Lead the design, implementation, rollout, and iteration of Baseten’s identity and authorization platform. Build fine-grained authorization, credential systems, identity lifecycle capabilities, agent credentials, and enterprise admin controls. Ensure low-latency permission checks, consistency, security, observability, and scalability across multi-tenant systems. Partner with product, design, and customer-facing teams, establish engineering practices, and provide technical mentorship as the team grows.

As a Forward Deployed Engineer, you will develop and maintain ML solutions, collaborate with teams on AI/ML pipelines, and manage projects end-to-end.

Design and implement RDMA/InfiniBand networking primitives for distributed GPU inference, optimize communication for Disaggregated KV Cache and WideEP, enable sub-10s model startup, validate hardware performance on H100/H200/B200/B300/GB300 clusters, build observability tools, and optimize or author communication kernels (NCCL/NVSHMEM/UCX).

As a Software Engineer, you'll build and maintain Baseten's core product, design APIs, implement features, fix issues, and collaborate with teams.

Develop and maintain components for machine learning inference platform. Focus on Kubernetes deployments, resource management, and performance monitoring.

Build and maintain responsive, accessible frontend interfaces using React and TypeScript. Collaborate with product, design, and backend teams to launch performant, scalable UIs, create reusable component libraries, define APIs, mentor engineers, and drive frontend best practices.

Build an internal AI developer platform that provisions agent configs, context tooling, eval harnesses, and rollout mechanics to make AI-assisted engineering the default. Integrate LLMs and third-party coding tools, enable self-serve agent creation, embed safety and audit controls, and measure impact on developer productivity.

The GPU Kernel Engineer will design and implement high-performance GPU kernels, optimize code, and collaborate with research teams to enhance AI workloads.

Lead development of enterprise platform features focusing on infrastructure, security, and compliance for enterprise customers. Oversee projects, collaborate cross-functionally, and mentor engineering teams.

As a Site Reliability Engineer, you'll build and maintain infrastructure for ML models, automate processes, and collaborate cross-functionally.

Build and evolve scalable observability systems: high-throughput telemetry ingest, cost-efficient storage, instrumentation libraries, SLO/alerting infrastructure, and integrations to improve reliability, diagnostics, and incident response across multi-cloud infrastructure.

The Software Engineer will lead the development of Baseten Voice AI, focusing on real-time model serving systems, collaboratng with teams on architecture and implementation.

As a Senior Software Engineer, you will manage Baseten's billing and revenue infrastructure, ensuring reliability and collaboration with finance and sales teams while evolving the billing platform.

Lead and grow a systems engineering team building container runtime and storage layers for production AI inference. Set technical direction, drive containerd/runc enhancements, own startup performance and weight/image delivery, and partner cross-functionally while contributing upstream to open-source container projects.

Partner with Sales and customers to discover requirements, design AI inference solutions, lead technical demos, scope deployments, run benchmarks, and execute proofs of concept. Own repeatable deployment playbooks across language, image, video, embedding, and voice models while advising on runtime, GPU, latency, and throughput tradeoffs. Coordinate Sales, Engineering, and Forward Deployed Engineering stakeholders to deliver customer implementations.