Member of Technical Staff - ML Research

Causal Labs • San Francisco, CA

2d

About The Position

This role involves working across the full machine learning stack, including data, model, evaluation, and infrastructure. The successful candidate will implement novel model architectures and training algorithms, and build data pipelines and training infrastructure for massive, petabyte-scale, multimodal datasets. The position requires rapid iteration on experiments and ablations, and staying up-to-date on research to bring new ideas to the work.

Requirements

• Strong grasp of machine learning fundamentals
• Depth in at least one core domain (e.g. Computer Vision, Sensor Fusion, Language Models, Physics-informed NNs)
• Experienced at training models and understanding experiment results through careful analysis and ablation studies
• Experienced at writing and optimizing massive petabyte-scale data pipelines
• Familiarity with distributed training

Nice To Haves

• Familiarity with meteorology, computational fluid dynamics, and/or numerical simulations

Responsibilities

• Work across the full ML stack (data, model, eval, and infrastructure)
• Implement novel model architectures and training algorithms
• Build data pipelines and training infrastructure for massive, petabyte-scale, multimodal datasets
• Rapidly iterate on experiments and ablations
• Stay up-to-date on research to bring new ideas to work

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed