Software Engineer – ML Infrastructure

Posted Yesterday
Be an Early Applicant
San Francisco, CA, USA
In-Office
150K-350K Annually
Senior level
Agency • Professional Services • Consulting
The Role
Build and operate infrastructure supporting machine learning experimentation, training, evaluation, and production serving. Responsibilities include developing reliable pipelines and shared tools, improving distributed compute utilization and scheduling, profiling and monitoring performance, automating testing, and making architectural decisions. The role requires ownership of ML infrastructure over time, collaboration with researchers and engineers, and expertise in distributed systems, GPU workloads, PyTorch, model serving, reinforcement learning post-training, RLHF, and model evaluation.
Summary Generated by Built In

Note: We are recruiting on behalf of our valued client. This opportunity is for a position with their organization, not with People Culture Talent. We're excited to help connect talented professionals with this exceptional team!

The Role

We’re seeking an experienced ML Infrastructure Software Engineer to build and improve infrastructure for machine learning workloads. You’ll partner with researchers and engineers to support experimentation, training, evaluation, and production deployment, with a focus on reliability and performance.

 
What You’ll Own
  • Design and build infrastructure for model training and production serving.

  • Develop reliable pipelines and shared tools that help teams run and manage machine learning workloads.

  • Improve compute utilization, workload scheduling, and performance across distributed systems.

  • Use profiling and monitoring to identify bottlenecks, troubleshoot failures, and guide improvements.

  • Automate testing and measurement to help teams assess changes consistently.

  • Lead architectural decisions, explain trade-offs, and carry projects from initial design through deployment and ongoing operation.

  • Collaborate with researchers and engineering teams to turn evolving requirements into maintainable systems.

What You’ll Bring
  • Substantial software engineering experience with technical ownership at Senior Engineer level or equivalent.

  • A track record of personally designing, building, and operating core machine learning infrastructure.

  • Hands-on experience with PyTorch, model serving engines, and distributed training frameworks

  • Strong experience with distributed systems and large-scale compute environments, including GPU-based workloads.

  • Solid background in reinforcement learning (RL) post-training infrastructure, RLHF, or automated model evaluation pipelines

  • Practical experience with modern machine learning frameworks and their supporting infrastructure.

  • Strong systems design, debugging, and performance optimization skills.

  • Clear, concise communication and the ability to explain your contributions, architectural choices, and technical trade-offs.

  • Experience owning systems over time, including maintenance, reliability, and improvements as requirements evolve.

Preferred experience
  • Experience supporting both training infrastructure and production model deployment.

  • Experience with large language model workloads is a plus, but not required. Infrastructure experience with other types of machine learning models is equally relevant.

Skills Required

  • Substantial software engineering experience with technical ownership at Senior Engineer level or equivalent
  • Experience designing, building, and operating core machine learning infrastructure
  • Hands-on experience with PyTorch, model serving engines, and distributed training frameworks
  • Strong experience with distributed systems and large-scale compute environments, including GPU-based workloads
  • Experience with reinforcement learning post-training infrastructure, RLHF, or automated model evaluation pipelines
  • Experience with modern machine learning frameworks and supporting infrastructure
  • Strong systems design, debugging, and performance optimization skills
  • Clear communication and ability to explain technical contributions, architectural choices, and trade-offs
  • Experience owning systems through maintenance, reliability improvements, and evolving requirements
  • Experience supporting both training infrastructure and production model deployment
  • Experience with large language model workloads
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
220 Employees
Year Founded: 2018

What We Do

Helping VCs and venture-backed startups build thriving teams and inspire their people through exceptional Human Capital Management. We specialize in creating people-first talent strategies for high-growth companies.

Similar Jobs

Philo Logo Philo

Software Engineer

Cloud • Digital Media • News + Entertainment • On-Demand
Remote or Hybrid
2 Locations
165 Employees
165K-250K Annually

Snap Inc. Logo Snap Inc.

Software Engineer

Artificial Intelligence • Cloud • Machine Learning • Mobile • Software • Virtual Reality • App development
Hybrid
3 Locations
5000 Employees
178K-313K Annually

Uber Logo Uber

Senior Software Engineer

Logistics • Transportation • 3PL: Third Party Logistics
In-Office
4 Locations
21000 Employees
202K-224K Annually

Pramana Labs Logo Pramana Labs

Software Engineer

Artificial Intelligence • Security • Software
In-Office
2 Locations
33 Employees

Similar Companies Hiring

Cravath, Swaine & Moore LLP Thumbnail
Legal Tech • Other • Professional Services
New York, NY
1200 Employees
VCA Animal Hospitals Thumbnail
Pet • Professional Services
Los Angeles, CA
35000 Employees
RapidScale Thumbnail
Cloud • Information Technology • Consulting
Raleigh, NC
600 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account