← All roles
Software Engineer - Training Infrastructure
GoPythonKubernetesAWSGCPDistributed SystemsTemporalApache AirflowPyTorch
About the role
As a Software Engineer on the Training Infrastructure team, you will architect and lead development of the training platform, supporting top tier research engineers and model developers. You will make key technical decisions for infrastructure enabling developers to deploy, scale, and monitor workloads with high performance and reliability. You will own scheduling, storage, networking, reliability, and observability of technical systems in the training stack. Responsibilities include designing a
Requirements
Bachelor’s degree or high in Computer Science or related field. Proficiency in Go, with Python experience a plus. Deep expertise with Kubernetes in production environments. Extensive experience with major cloud providers (AWS, GCP) and neo-cloud providers (Crusoe, DigitalOcean, Nebius) a plus. Advanced understanding of distributed systems concepts and performance tuning. Proven experience designing observability systems. Experience with ML/AI workloads and MLOps platforms highly valued.
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
Engineering Manager, Runtime Fabric
GoLinuxKubernetesDistributed Systems
$165k–330k
Onsite · San Francisco
Engineering Manager, Runtime Fabric
$165k–330k
GoLinuxKubernetesDistributed Systems
OS / K8s Systems Engineer
KubernetesPythonGoLinux
$165k–330k
Onsite · San Francisco
OS / K8s Systems Engineer
$165k–330k
KubernetesPythonGoLinux
Software Engineer - Internal Platform
GoPythonKubernetesDockerHelmTerraform
$165k–330k
Remote · CA, US
Software Engineer - Internal Platform
$165k–330k
GoPythonKubernetesDockerHelmTerraform
Software Engineer - Infrastructure
PythonGoKubernetesDistributed Systems
$165k–330k
Remote · CA, US
Software Engineer - Infrastructure
$165k–330k
PythonGoKubernetesDistributed Systems
Product Engineer - Dedicated Inference
PythonGoJavaScript
$165k–330k
Onsite · San Francisco
Product Engineer - Dedicated Inference
$165k–330k
PythonGoJavaScript
Similar roles
Staff Software Engineer, Platform Infrastructure (Foundations)
$215k–265k
Staff Software Engineer, Platform Infrastructure (Foundations)
$215k–265k
Staff Software Engineer (Hardware Infrastructure)
$175k–225k
Staff Software Engineer (Hardware Infrastructure)
$175k–225k
Staff+ Software Engineer, Platform
$405k–485k
Staff+ Software Engineer, Platform
$405k–485k
Staff Software Engineer, ML Infrastructure
$147k–215k
Staff Software Engineer, ML Infrastructure
$147k–215k
Member of Technical Staff - Distributed Systems
$150k–350k
Member of Technical Staff - Distributed Systems
$150k–350k