Senior Staff Machine Learning Engineer

What you'll do

• Design and build infrastructure for petabyte-scale data ingestion, versioning, and lineage across multimodal datasets.
• Develop Python SDKs and core libraries to help ML engineers transform and load model-ready data.
• Build high-throughput data loading primitives to ensure GPU workloads remain compute-bound rather than I/O-bound.
• Operate distributed data pipelines using Spark, Daft, and Rust-based systems for large-scale preparation.
• Optimize platform components for seamless integration with PyTorch, JAX, and TensorFlow frameworks.
• Implement governance features including legal terms enforcement, privacy controls, and end-to-end data lineage.
• Support next-generation workloads such as foundation models, multimodal data, and retrieval-augmented systems.
• Diagnose and automate complex issues across the stack to maximize system uptime and throughput.

What we're looking for

• A B.S., M.S., or Ph.D. in Computer Science, Computer Engineering, or equivalent practical experience is required.
• Candidates must have a strong foundation in machine learning and hands-on experience across the end-to-end ML workflow.
• Experience building and running large-scale distributed systems and maintaining scalable, highly available systems is required.
• Proven experience delivering data and machine learning infrastructure in real-world production environments is required.
• Candidates must have extensive programming experience in Java, Python, or Go.
• Familiarity with modern generative techniques, including transformers, diffusion, and retrieval-augmented generation, is required.
• Experience with fine-tuning workflows, model optimization, and preparing models for scalable inference is required.
• Proficiency with ML frameworks like PyTorch, JAX, or TensorFlow is preferred.