Baseten Logo

Baseten

Software Engineer - Infrastructure

Reposted 20 Days Ago
Remote or Hybrid
Hiring Remotely in New York, NY, USA
165K-330K Annually
Junior
Remote or Hybrid
Hiring Remotely in New York, NY, USA
165K-330K Annually
Junior
Develop and maintain components for machine learning inference platform. Focus on Kubernetes deployments, resource management, and performance monitoring.
The summary above was generated by AI

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 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:

  • 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

  • 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

  • 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).

Baseten New York, New York, USA Office

New York, NY, United States

Similar Jobs

Yesterday
Easy Apply
Remote
Easy Apply
181K-241K Annually
Senior level
181K-241K Annually
Senior level
Big Data • Fintech • Mobile • Payments • Financial Services
Lead technical strategy and execution for the shared mobile app infrastructure, build and maintain React Native and native modules, improve observability, CI/CD, releases and developer tooling, define standards and runbooks, mentor engineers, coordinate cross-functional trade-offs, and ensure production availability and quality across mobile products.
Top Skills: AppiumClaude CodeCrashlyticsExpoJavaScriptLaunchdarklyMaestroReact NativeRedux ToolkitSentryStatsigTypescript
11 Days Ago
Easy Apply
Remote
Easy Apply
153K-213K Annually
Senior level
153K-213K Annually
Senior level
Big Data • Fintech • Mobile • Payments • Financial Services
Lead and deliver backend batch infrastructure work: own quarterly goals, design and build reliable scalable compute for batch workloads, collaborate with product and stakeholders, run ops/on-call, set code/design standards, mentor engineers, and drive platform reliability and performance.
Top Skills: AirflowAWSFlinkFlyteKotlinKubernetesLuigiMySQLPrefectPythonSparkTemporal
4 Days Ago
Remote
153K-189K Annually
Senior level
153K-189K Annually
Senior level
Financial Services
Build and operate Wealthsimple's ML platform to take models from experimentation to production: design architecture, build framework-agnostic training/inference services, ensure scalability and observability, enable self-serve tooling for data scientists, and own long-lived ML infra projects (Kubernetes, Ray, LLM integrations).
Top Skills: BedrockChronon Feature StoreCudaDvcElasticsearchFastapiKafkaKubernetesMlflowPostgresPythonPyTorchRayRedshiftSnowflakeTransformersTritonVector DatabasesVllm

What you need to know about the NYC Tech Scene

As the undisputed financial capital of the world, New York City is an epicenter of startup funding activity. The city has a thriving fintech scene and is a major player in verticals ranging from AI to biotech, cybersecurity and digital media. It also has universities like NYU, Columbia and Cornell Tech attracting students and researchers from across the globe, providing the ecosystem with a constant influx of world-class talent. And its East Coast location and three international airports make it a perfect spot for European companies establishing a foothold in the United States.

Key Facts About NYC Tech

  • Number of Tech Workers: 549,200; 6% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Capgemini, Bloomberg, IBM, Spotify
  • Key Industries: Artificial intelligence, Fintech
  • Funding Landscape: $25.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Greycroft, Thrive Capital, Union Square Ventures, FirstMark Capital, Tiger Global Management, Tribeca Venture Partners, Insight Partners, Two Sigma Ventures
  • Research Centers and Universities: Columbia University, New York University, Fordham University, CUNY, AI Now Institute, Flatiron Institute, C.N. Yang Institute for Theoretical Physics, NASA Space Radiation Laboratory

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account