Senior ML Researcher — 3D Geology & Diffusion Models

Terra AI; Palo Alto (CA); On-site; USD 100,000 - 160,000; Full time

Job summary

Terra AI is seeking a talented deep learning engineer or scientist in Palo Alto, California to lead the development of generative models for 3D geological structures and decision-making in clean energy applications.

The ideal candidate will have extensive experience with PyTorch, a strong foundation in developing large deep learning architectures, and skills in data curation. Responsibilities include designing and iterating on diffusion models, working with geophysical data, and collaborating with project teams.

Qualifications

• Extensive experience with PyTorch, including custom module creation.
• Ability to design and implement large-scale deep learning models.
• Hands-on skills in creating and maintaining high-quality datasets.
• Proficient in software development best practices.

Responsibilities

• Design and iterate diffusion models for 3D geological models.
• Condition generation on geophysical data and other observations.
• Inform synthetic data generation for model performance.
• Collaborate with project teams for specific real-world projects.

Skills

PyTorch

Deep Learning

Data Curation

Software Engineering

Job description

Terra AI is seeking a talented deep learning engineer or scientist in Palo Alto, California to lead the development of generative models for 3D geological structures and decision-making in clean energy applications.

The ideal candidate will have extensive experience with PyTorch, a strong foundation in developing large deep learning architectures, and skills in data curation. Responsibilities include designing and iterating on diffusion models, working with geophysical data, and collaborating with project teams.