ML Inference Platform Engineer

Baseten; San Francisco (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Competitive equity package

100% medical, dental, and vision for U

Flexible PTO including Winter Break

Parental leave

Fertility and family-building stipend

401(k) plan (U.S. employees)

Job summary

Baseten is building an ML inference platform powering production AI apps. As an Infrastructure Software Engineer, you will develop core components, deploy and monitor model serving, and optimize resource usage with Go and Python in a Kubernetes-driven environment.

You will collaborate across teams to improve reliability, scalability, and performance while advancing our orchestration layer and monitoring capabilities for cutting-edge AI workloads.

Qualifications

• Bachelor's degree in Computer Science or related field.
• Proficient coding abilities in one or more popular programming languages; Go proficiency is a plus.
• Working knowledge of Kubernetes and containerization.
• Basic understanding of machine learning concepts and model serving.
• Familiarity with distributed systems concepts.
• Experience with basic monitoring and logging tools.
• Interest in ML/AI infrastructure and willingness to learn.
• Strong collaboration and communication skills.

Responsibilities

• Develop infrastructure components for our ML inference platform using Python and Go.
• Implement and maintain Kubernetes deployments for model serving.
• Contribute to our inference orchestration layer for model deployments.
• Build and enhance monitoring systems for model performance metrics.
• Implement efficient resource management solutions for ML workloads.
• Support infrastructure automation to improve ML deployment workflows.
• Work closely with team members to implement technical solutions.
• Help balance performance optimization with system reliability.
• Participate in technical discussions around infrastructure improvements.
• Learn and apply infrastructure best practices.

Skills

Go

Distributed systems

Monitoring tools

Collaboration

Go proficiency is a plus

Education

Bachelor's degree in Computer Science or related

Tools

Kubernetes

Job description

Baseten is building an ML inference platform powering production AI apps. As an Infrastructure Software Engineer, you will develop core components, deploy and monitor model serving, and optimize resource usage with Go and Python in a Kubernetes-driven environment.

You will collaborate across teams to improve reliability, scalability, and performance while advancing our orchestration layer and monitoring capabilities for cutting-edge AI workloads.