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' mission is to build safe, steerable AI physics models, starting with accurate weather prediction and control.Our team is small and driven by deep passion and a sense of urgency for solving the most important problems facing humanity. We've previously built and led the development of safety-critical models in robotics and autonomous vehicles. We're funded by leading investors in Silicon Valley and are lucky to have the support of amazing advisors &amp; angels from institutions like OpenAI, Google DeepMind, Cruise, Waymo, and many more.

Member of Technical Staff - ML Infra

About the Job

About the Company

Causal Labs

Similar Jobs

Senior Software Engineer, Machine Learning Infrastructure

Senior Software Engineer, Machine Learning Infrastructure

Staff Computer Vision Engineer - vLLM

Staff Computer Vision Engineer - vLLM

Senior Staff Computer Vision Engineer - Deployment

Senior Staff Computer Vision Engineer - Deployment