A geoscience frontier company building AI to accelerate subsurface resource development for a clean energy future. Backed by Khosla.
About the role
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
What they're looking for
- 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
More about this role
Terra AI is building a new category at the intersection of artificial intelligence, geoscience, and critical resource development.
As global demand for copper, lithium, nickel, rare earth elements, geothermal energy, and other strategic resources accelerates, the mining and subsurface industries face a growing challenge: traditional exploration methods remain slow, expensive, and highly uncertain. Terra AI was founded to help solve this problem by redefining how critical resources are discovered, evaluated, and developed.
By combining advanced machine learning, probabilistic modeling, and deep geoscience expertise, Terra AI helps exploration and mining companies make faster, more informed subsurface decisions with greater confidence and capital efficiency. The company’s platform integrates geological, geophysical, and drilling data into intelligent systems designed to improve targeting accuracy, accelerate discovery timelines, and reduce exploration risk.
Backed by leading investors including Khosla Ventures and working alongside strategic industry partners including Rio Tinto, Ero Copper, and Ramaco Resources, Terra AI is emerging as one of the more closely watched AI-native...
Browse similar: AI jobs · AI startup jobs · Startup jobs · Remote jobs