Jobs at Baseten

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.

Lead and mentor a team of Forward Deployed Engineers in building and optimizing LLM inference workloads, delivering high-performance AI applications, and collaborating with cross-functional teams.

The role involves developing and managing infrastructure for distributed LLM inference systems, ensuring performance and usability, debugging complex systems, and collaborating with performance engineers.

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.

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

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

As an AI Support Engineer, you'll manage support requests, resolve user issues, optimize ML models, and contribute to product development.

Partner with Sales and customers to run technical discovery, lead demos and scoping, design repeatable ML inference deployments across modalities, benchmark inference performance, and drive POC execution and stakeholder alignment.

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.

Partner with customers to architect, build, and deploy high-scale production AI applications. Translate ambiguous goals into reliable, observable services, implement end-to-end model deployment and monitoring, optimize ML projects, and own customer-facing technical delivery and product outcomes.

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.