Software Engineer - Infrastructure

Baseten; New York (NY); On-site; USD 165,000 - 330,000; Full time

Benefits offered by this job

Competitive compensation, including equity

100% coverage of medical, dental, and vision insurance

Generous PTO policy

Paid parental leave

Company-facilitated 401(k)

Job summary

A leading AI company in New York is seeking an Infrastructure Software Engineer to develop and maintain components of their ML inference platform. Responsibilities include implementing Kubernetes deployments, enhancing monitoring systems, and developing infrastructure components using Python and Go. The ideal candidate should have a bachelor's degree in Computer Science, coding proficiency, and a working knowledge of Kubernetes. The role offers competitive compensation ranging from $165K to $330K and generous benefits including medical, dental, and vision insurance.

Qualifications

• Bachelor’s degree or higher in Computer Science or related field.
• Proficient coding abilities in one or more popular programming or scripting languages.
• Working knowledge of Kubernetes and containerization.
• Basic understanding of machine learning concepts and model serving.
• Familiarity with distributed systems concepts.

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.
• Support infrastructure automation to improve ML deployment workflows.

Skills

Proficient coding abilities

Knowledge of Kubernetes

Interest in ML/AI infrastructure

Strong collaboration skills

Education

Bachelor’s degree in Computer Science or related field

Tools

Monitoring and logging tools

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 $300M Series E, backed by investors including BOND, IVP, Spark Capital, Greylock, and Conviction. Join us and help build the platform engineers turn to 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

• Multi‑cloud capacity management
• Inference on B200 GPUs
• Multi‑node inference
• Fractional H100 GPUs for efficient model serving

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.
• 100% coverage of medical, dental, and vision insurance for employee and dependents
• Generous PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year’s Day!)
• Paid parental leave
• 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).

Compensation Range: $165K – $330K