Member of Technical Staff - Machine Learning Infrastructure Engineer

Preference Model

$150K — $200K *
Technical Services
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • 5+ years of software engineering experience, particularly in production-grade infrastructure
  • Proficient in core machine learning frameworks like PyTorch or JAX
  • Hands-on experience with cloud services (AWS, GCP) and container orchestration (Kubernetes)
  • Experience building robust scalable data pipelines and engineering tools
  • Familiarity with LLM training and inference internals is a plus
  • Strong problem-solving skills to bridge the gap between research needs and infrastructure capabilities
  • Effective communicator able to convey technical concepts to non-specialists

Responsibilities

  • Design and scale compute, scheduling, and data infrastructure for ML research
  • Develop and maintain essential ML framework primitives and internal tools for reproducible experimentation
  • Build evaluation and monitoring infrastructure to ensure reliability and early failure detection
  • Partner with research engineers to translate their needs into actionable infrastructure
  • Implement automated testing and deployment systems for efficient infrastructure management

Benefits

  • Competitive cash and equity compensation
  • Ownership and autonomy in a dynamic startup environment
  • Collaboration with experienced engineers from leading labs
  • Comprehensive health, vision, and dental benefits
  • 401K matching
  • Daily onsite lunch and weekly snack orders
  • Visa sponsorship and relocation support
Full Job Description
About the Role

Frontier research moves only as fast as its infrastructure permits. Building solid infrastructure is foundational to our mission of pushing self-directed learning as far as it can go.

We are looking for ML Infrastructure Engineers to build the systems that power the frontier of post-training on large language models. This role involves building scalable infrastructure to enable high-throughput systems and shape how our research is run, bringing us closer to models that can train themselves on what they aren't yet good at.

What You Will Do
  • Design, build, and scale the compute, scheduling, and data infrastructure that powers post-training research on our in-house RL environments
  • Develop and maintain core ML framework primitives and internal tooling that researchers rely on daily, accelerating reproducible experimentation and reducing time from idea to result
  • Build evaluation and benchmarking infrastructure, monitoring, logging, and debugging tooling, and automated testing and deployment systems, so failures are caught early and infrastructure stays reliable as it scales
  • Partner directly with Research Engineers to translate research needs into infrastructure requirements, and ship fast in response to their feedback
What We are Looking For
  • Have strong software engineering fundamentals, experience building production-grade infrastructure (ideally for ML or data-intensive systems), and proficiency in core ML frameworks such as PyTorch or JAX
  • Understand distributed systems principles, and have hands-on experience with cloud platforms (AWS, GCP) and container orchestration (Kubernetes), building systems for high-throughput, low-latency workloads
  • Have experience with data engineering tools and building robust, scalable data pipelines
  • Have some familiarity with LLM training/inference internals (transformers, distributed training, inference libraries like vLLM or SGLang) - deep expertise is a plus, not a requirement
  • Can balance production rigor with the pace of fast-moving research, and communicate infrastructure tradeoffs clearly to researchers who aren't infra specialists


What We Offer:
  • Competitive cash and equity compensation (>90th percentile)
  • Ownership and autonomy in a fast moving startup environment
  • Opportunity to work alongside senior and staff engineers from frontier labs and infrastructure companies, plus top ML engineers
  • Health, vision, dental, benefits
  • 401K match
  • Lunch provided everyday onsite
  • Weekly snack orders
  • Visa sponsorship & relocation support available


We value diverse perspectives and experiences. If you're excited about this role but don't check every box, we still encourage you to apply.

Similar Jobs

More Jobs at Preference Model

More Technical Services Jobs

Find similar Member of Technical Staff - Machine Learning Infrastructure Engineer jobs: