Terra AI

Senior/Staff Machine Learning Researcher

US remote · Staff+

Sponsorship not specifiedDetected 99 days ago

Cloud PlatformsMachine LearningDeep LearningPyTorchNLPCommunicationCollaboration

About the role

The outputs of the generative model capture what we know and don't know about the state of the subsurface, allowing explorers to make maximally informed decisions about how and where to explore for critical resources.

Design, train, test, and iterate on diffusion models for 3D geological models
Design, train, test, and iterate on an approach for conditioning generation on geophysical data and other observations

Familiarity with designing scalable and maintainable systems.
Experience with Generative Models
Familiarity with generative architectures, particularly diffusion models, and an emphasis on posterior sampling methods.
Knowledge of Transformer Architectures
Expertise in parallelizing models across multiple GPUs and optimizing distributed training pipelines.
Cloud Infrastructure Expertise
Required Qualifications:
Extensive PyTorch Experience
Deep understanding of PyTorch, including writing custom modules, optimizing training, and debugging issues in large-scale models.
Expertise in Developing Large Deep Learning Models from Scratch
Proven ability to design, implement, and train complex deep learning architectures from the ground up.
Data Curation Skills
Hands-on experience in creating, cleaning, and maintaining high-quality datasets tailored for machine learning applications.
Strong Software Engineering and Design Experience
Proficient in software development best practices, including version control, testing, and code optimization.
Nice-to-haves:

We are looking for a talented deep learning engineer or scientist to lead the development of this model that will revolutionize decision-making in the earth subsurface for a wide range of clean energy applications.

This listing is sourced directly from Terra AI's careers page and normalized into a canonical job model.