Senior Foundation Models Engineer

Coastal Carbon; San Francisco (CA); Hybrid; USD 150,000 - 200,000; Full time

Job summary

Coastal Carbon is looking for a Senior Machine Learning Engineer to develop cutting-edge AI models that understand our planet. This position requires a strong background in Python and PyTorch, with 5+ years of experience in building distributed training systems.

The ideal candidate will be heavily involved in designing, implementing, and scaling state-of-the-art models while collaborating with researchers. Preference for candidates in San Francisco or Waterloo, but remote is possible for outstanding applicants.

Qualifications

• 5+ years of relevant work experience.
• Prior experience building distributed training pipelines for multi-node systems using PyTorch and Ray.
• Experience training large diffusion or transformer models, preferably on video or time series data.

Responsibilities

• Designing, implementing, and scaling state-of-the-art models.
• Productionizing research codes and complex systems.
• Develop AI systems that understand the universe and generate knowledge.

Skills

Python

PyTorch

Ray Trainer

Distributed training pipelines

Cloud platforms (AWS, GCP, Azure)

Education

Bachelor’s degree in computer science, engineering, or related field

Tools

Ray

Anyscale framework

Job description

Coastal Carbon is looking for a Senior Machine Learning Engineer to develop cutting-edge AI models that understand our planet. This position requires a strong background in Python and PyTorch, with 5+ years of experience in building distributed training systems.

The ideal candidate will be heavily involved in designing, implementing, and scaling state-of-the-art models while collaborating with researchers. Preference for candidates in San Francisco or Waterloo, but remote is possible for outstanding applicants.