Lead Diffusion ML Engineer for 3D Geology Models

terraai; United States; Remote; USD 130,000 - 210,000; Full time

Job summary

Terra AI is seeking a talented deep learning engineer or scientist to lead the development of a generative model that produces 3D geological representations conditioned on geophysical surveys and borehole measurements. The role focuses on diffusion modeling and applying ML to critical resource exploration.

You will design and train large DL models from scratch, curate high-quality datasets, and collaborate with project teams on real-world mining and clean energy applications, including scalable

Qualifications

• Extensive PyTorch experience with custom modules, optimizing training, and debugging large-scale models.
• Ability to design, implement, and train complex deep learning architectures from scratch.
• Hands-on data curation: creating, cleaning, and maintaining high-quality datasets for ML.
• Strong software engineering practices: version control, testing, and scalable system design.
• Experience with generative models, transformers, and distributing training across GPUs and clouds.

Responsibilities

• Design, train, test, and iterate on diffusion models for 3D geological models
• Design, train, test, and iterate on conditioning generation on geophysical data and other observations
• Inform the generation of synthetic data to improve model performance
• Adapt diffusion modeling approach to real-world projects in collaboration with project teams

Skills

PyTorch

DL Architectures

Data Curation

Software Engineering

Version Control

Generative Models

Transformers

Distributed Training

Cloud Infrastructure

Job description

Terra AI is seeking a talented deep learning engineer or scientist to lead the development of a generative model that produces 3D geological representations conditioned on geophysical surveys and borehole measurements. The role focuses on diffusion modeling and applying ML to critical resource exploration.

You will design and train large DL models from scratch, curate high-quality datasets, and collaborate with project teams on real-world mining and clean energy applications, including scalable