ML Data Infrastructure Engineer

As a member of our ML Data Platform team, you'll solve technical challenges, including upgrading and implementing state-of-the-art software infrastructure. The team builds a high-performance, high availability, globally distributed ecosystem platform of services that in turn provide the foundation for rapid development of novel new systems that integrate into that ecosystem and improve it.

The Impact You'll Make

• Design and build data processing infrastructure for model training and feature serving, optimizing for performance, reproducibility, and traceability
• Collaborate closely with research teams to design and implement novel data processing architectures for emerging model and training paradigms
• Identify and resolve performance bottlenecks across the training data pipeline, from raw data ingestion to feature delivery
• Establish best practices, tooling for data infrastructure used across ML teams

Required Qualifications

• Have 1 - 3 years of experience and a minimum of a BS and/or MS in Computer Science
• Strong software engineering fundamentals, with experience building high-throughput, fault-tolerant distributed systems
• Hands-on experience with distributed computing frameworks such as Apache Spark or Flink
• Solid grounding in data structures, systems design, and performance optimization
• Strong problem-solving skills and attention to detail

Preferred Qualifications

• Background in MLOps, Data Infrastructure, or ML Infrastructure
• Experience with ML training pipelines, feature stores, or model-serving systems

Other Types of Pay: Equity eligible

Paid Sick Leave: 80 hours per year

Method of Application: Apply online

CA Base Pay Range $150,000 — $224,000 USD