Responsibilities<ul><li>Design, deploy, and maintain large distributed ML training and inference clusters</li><li>Develop efficient, scalable end-to-end pipelines to manage petabyte-scale datasets and model training throughout the entire ML lifecycle</li><li>Research and test various training approaches including parallelization techniques and numerical precision trade-offs across different model scales</li><li>Analyze, profile and debug low-level GPU operations to optimize performance</li><li>Stay up-to-date on research to bring new ideas to work</li></ul>What we’re looking forWe value a relentless approach to problem-solving, rapid execution, and the ability to quickly learn in unfamiliar domains.<ul><li>Strong grasp of state-of-the-art techniques for optimizing training and inference workloads</li><li>Demonstrated proficiency with distributed training frameworks (e.g. FSDP, DeepSpeed) to train large foundation models</li><li>Knowledge of cloud platforms (GCP, AWS, or Azure) and their ML/AI service offerings</li><li>Familiarity with containerization and orchestration frameworks (e.g., Kubernetes, Docker) </li><li>Background working on distributed task management systems and scalable model serving &amp; deployment architectures</li><li>Understanding of monitoring, logging, observability, and version control best practices for ML systems</li></ul>You don’t have to meet every single requirement above.

Causal Labs is a research company building AI that is capable of (1) predicting the future and (2) identifying the optimal actions to change that future.We call this general causal intelligence. To achieve this breakthrough, we are building a Large Physics foundation Model (LPM) to learn causality through physics and weather.We are a team of researchers and engineers from self-driving, drug discovery, and robotics who believe general causal intelligence will be the most important technical breakthrough for civilization.

Member of Technical Staff - ML Infra

Responsibilities<ul><li>Design, deploy, and maintain large distributed ML training and inference clusters</li><li>Develop efficient, scalable end-to-end pipelines to manage petabyte-scale datasets and model training throughout the entire ML lifecycle</li><li>Research and test various training approaches including parallelization techniques and numerical precision trade-offs across different model scales</li><li>Analyze, profile and debug low-level GPU operations to optimize performance</li><li>Stay up-to-date on research to bring new ideas to work</li></ul>What we’re looking forWe value a relentless approach to problem-solving, rapid execution, and the ability to quickly learn in unfamiliar domains.<ul><li>Strong grasp of state-of-the-art techniques for optimizing training and inference workloads</li><li>Demonstrated proficiency with distributed training frameworks (e.g. FSDP, DeepSpeed) to train large foundation models</li><li>Knowledge of cloud platforms (GCP, AWS, or Azure) and their ML/AI service offerings</li><li>Familiarity with containerization and orchestration frameworks (e.g., Kubernetes, Docker) </li><li>Background working on distributed task management systems and scalable model serving &amp; deployment architectures</li><li>Understanding of monitoring, logging, observability, and version control best practices for ML systems</li></ul>You don’t have to meet every single requirement above.

Member of Technical Staff - ML Infra

About the Job

About the Company

Causal Labs

Similar Jobs

Senior Robotics AI Engineer

Senior Robotics AI Engineer

Staff Machine Learning Engineer - AI Foundation

Staff Machine Learning Engineer - AI Foundation

Principal AI Research Engineer

Principal AI Research Engineer

Senior Data Engineer

Senior Data Engineer

Lead, AI Engineering

Lead, AI Engineering

Machine Learning Engineer-Technical Lead

Machine Learning Engineer-Technical Lead

Senior AI Systems Engineer

Senior AI Systems Engineer

Staff AI Systems Engineer

Staff AI Systems Engineer

Staff Engineer, AI Labs

Staff Engineer, AI Labs

Staff Machine Learning Engineer

Staff Machine Learning Engineer