← All roles
Software Engineer - Training Infrastructure
GoKubernetesDistributed SystemsPyTorchApache AirflowTemporalAWSGCP
About the role
As a Software Engineer on the Training Infrastructure team at Baseten, you will architect and lead development of the training platform, supporting top-tier research engineers and model developers. You will own scheduling, storage, networking, reliability, and observability of technical systems in the training stack.
Requirements
Bachelor's degree or higher in Computer Science or related field. Proficiency in Go. Deep expertise with Kubernetes in production environments. Advanced understanding of distributed systems concepts and performance tuning. Proven experience designing observability systems. Experience with ML/AI workloads and MLOps platforms.
About the company
Baseten
Baseten provides the Baseten Inference Stack, a platform for high-performance AI model deployment and management across any cloud. They offer the fastest model runtimes, cross-cloud high availability, and seamless developer workflows to help businesses bring AI products to market quickly and efficiently.
View company page →More from Baseten
Global Capacity Manager
KubernetesGoPython
$170k–230k
Onsite · San Francisco
Global Capacity Manager
$170k–230k
KubernetesGoPython
Software Engineer - Identity & Authorization
KubernetesPythonGoOIDC
$240k–285k
Onsite · San Francisco
Software Engineer - Identity & Authorization
$240k–285k
KubernetesPythonGoOIDC
Software Engineer - AI Developer Productivity
PythonGoKubernetesDocker
$165k–330k
Onsite · San Francisco
Software Engineer - AI Developer Productivity
$165k–330k
PythonGoKubernetesDocker
Software Engineer - Testing Frameworks
GoPythonpytestk6KubernetesDocker
$165k–330k
Onsite · San Francisco
Software Engineer - Testing Frameworks
$165k–330k
GoPythonpytestk6KubernetesDocker
Software Engineer - Observability
PrometheusGrafanaClickHouseOTELPythonRustGo
$165k–330k
Onsite · San Francisco
Software Engineer - Observability
$165k–330k
PrometheusGrafanaClickHouseOTELPythonRustGo
Similar roles
Staff+ Software Engineer, Kubernetes Platform
$320k–485k
Staff+ Software Engineer, Kubernetes Platform
$320k–485k
Staff Cloud Software Engineer
zł351k–474k
Staff Cloud Software Engineer
zł351k–474k
Principal Engineer, Cluster Orchestration
$206k–303k
Principal Engineer, Cluster Orchestration
$206k–303k
Staff+ Software Engineer, Infrastructure (Distributed Systems)
$320k–485k
Staff+ Software Engineer, Infrastructure (Distributed Systems)
$320k–485k
Staff+ Software Engineer, Node Infra
$320k–485k
Staff+ Software Engineer, Node Infra
$320k–485k