Lead ML Engineer - 3D Geological Diffusion Models

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

Job summary

Terra AI in Palo Alto is seeking a talented deep learning engineer to pioneer the development of a generative model for 3D geological applications in the clean energy sector.

The role requires extensive experience with PyTorch and the ability to design and implement large-scale deep learning models. Responsibilities include training diffusion models, curating high-quality datasets, and collaborating with project teams.

Qualifications

• Extensive know-how in PyTorch, including custom modules and debugging.
• Proven experience in designing and training complex deep learning architectures.
• Hands-on skills in creating and maintaining high-quality datasets.
• Strong software engineering practices with code optimization expertise.

Responsibilities

• Design, train, and test diffusion models for 3D geological outputs.
• Develop methods conditioning generation on geophysical data.
• Generate synthetic data enhancing model performance.
• Adapt models for specific projects in collaboration with teams.

Skills

PyTorch Experience

Large Deep Learning Models

Data Curation Skills

Software Engineering

Job description

Terra AI in Palo Alto is seeking a talented deep learning engineer to pioneer the development of a generative model for 3D geological applications in the clean energy sector.

The role requires extensive experience with PyTorch and the ability to design and implement large-scale deep learning models. Responsibilities include training diffusion models, curating high-quality datasets, and collaborating with project teams.