Machine Learning Engineer - Infrastructure

Causal Labs is seeking a Machine Learning Engineer to design, deploy, and maintain large distributed ML training and inference clusters for a Large Physics foundation Model. The role involves building scalable end-to-end pipelines for petabyte-scale datasets and optimizing performance through low-level GPU operations and distributed training frameworks. This position requires deep expertise in large-scale ML infrastructure and the ability to work in person in San Francisco five days a week.