Remote Jobs RockRemote Jobs Rock

Software Engineer - Infrastructure

πŸ“… Mar 9, 2025
PythonGoKubernetesContainerization
Apply Now

πŸ“œ Description

  • Develop ML inference platform infrastructure using Python and Go.
  • Implement and maintain Kubernetes deployments for model serving.
  • Contribute to inference orchestration for model deployments.
  • Build monitoring systems for model performance metrics.
  • Improve resource management and automation for ML workloads and deployment workflows.
  • Collaborate on infrastructure improvements, performance optimization, and system reliability.

πŸ› οΈ Requirements

  • Bachelor's degree or higher in Computer Science or related field
  • 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

✨ Benefits

  • Competitive compensation, including meaningful equity
  • (U.S. only) 100% coverage of medical
  • Dental
  • Vision insurance for employee and dependents
  • Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
  • Paid parental leave
  • Fertility and family-building stipend through Carrot
  • (U.S. only) Company-facilitated 401(k)
  • Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.
Full job description

ABOUT BASETEN

Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma, and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F, led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to ship AI products.

THE ROLE

As an Infrastructure Software Engineer at Baseten, you'll build and maintain components of our ML inference platform that powers production AI applications. You'll contribute to the core infrastructure, enabling developers to deploy, scale, and monitor ML models with high performance.

EXAMPLE INITIATIVES

You'll get to work on these types of projects as part of our Infrastructure team:

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

REQUIREMENTS

  • Bachelor's degree or higher in Computer Science or related field

  • Proficient coding abilities in one or more popular programming or scripting 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

BENEFITS

  • Competitive compensation, including meaningful equity

  • (U.S. only) 100% coverage of medical, dental, and vision insurance for employee and dependents

  • Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)

  • Paid parental leave

  • Fertility and family-building stipend through Carrot

  • (U.S. only) Company-facilitated 401(k)

  • Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Apply now to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.

At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.

We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law (by example, the requirements of the San Francisco Fair Chance Ordinance, where applicable).

Niantic-Spatial

Software Engineer, Infrastructure

πŸ•’ 12 days ago
Niantic-Spatial🧠 AI-native

Build and operate cloud infrastructure, developer platforms, and edge services while improving reliability, enabling safe software delivery, and supporting compute and GPU capacity across engineering teams.

Infrastructure As CodeTerraformPulumiGCP
Zus Health

Infrastructure Engineer

πŸ•’ 20 days ago
Zus HealthπŸ‘₯ 51 - 200 employees🏒 Healthcare

Build and scale AWS infrastructure for a healthcare data platform, improving reliability, security, and cost efficiency while enabling engineering teams to deploy software and support production systems.

AWSTerraformCloudformationDocker
Thought-Machine

Software Engineer (Infrastructure)

πŸ•’ 12 days ago
Thought-Machine🏒 Financial Technology🀝 B2B

Build and maintain infrastructure tooling that automates deployments, improves service reliability and observability, and helps customers manage cloud-based banking systems.

PythonGolangInfrastructure Management ToolingObservability
Cohere

Software Engineer, GPU Infrastructure

πŸ•’ 5 days ago
CohereπŸ‘₯ 10,000+ employees🏒 Software Development🀝 B2B🧠 AI-native

Operate and improve multi-cloud GPU and TPU Kubernetes fleets for model training and serving, focusing on reliability, performance, utilization, automated fault recovery, and hardware partnerships.

KubernetesGPU InfrastructureTPU InfrastructureLinux
Designworkstalent

Virtualization & Orchestration Engineer - Member of Technical Staff

πŸ•’ yesterday
DesignworkstalentπŸ‘₯ 1 - 10 employees🏒 Staffing And Recruiting

Build and operate virtualization, Kubernetes orchestration, and provisioning systems that allocate GPU capacity reliably across multi-tenant AI infrastructure, from workload scheduling to cluster lifecycle management.

KubernetesContainer OrchestrationVirtualizationGPU Infrastructure

Trusted by Remote Workers