Senior/Staff Machine Learning Researcher at Terraai

Job description

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 Khosla Ventures, BHP, and Rio Tinto, Terra AI works with industry leaders to bring a more informed, data-driven approach to critical resource development. Terra AI’s mission is to define the new global standard for data-driven critical resource development — breaking the cost and time curve required to support electrification, energy security, and the global energy transition. The company operates with a strong partnership mentality, combining technical rigor, candid communication, continual learning, and environmental stewardship to help modern exploration teams solve some of the world’s most important resource challenges. Who you are You’re an AI/ML researcher who enjoys tackling problems without an established playbook. You’re excited by the challenge of developing novel generative models, testing new approaches, and using rigorous experimentation to understand what works, what fails, and why. You combine strong PyTorch skills with the ability to design and train substantial models from scratch. You know that meaningful research depends on thoughtful data curation, reproducible experiments, and software that others can understand and build upon. You enjoy collaborating across machine learning, geoscience, and engineering. You’re excited by the opportunity to apply generative AI to a genuinely difficult physical-world problem: understanding the geology we cannot directly see. Prior geoscience experience is welcome, but curiosity and a willingness to learn matter just as much. What you’ll do Terra AI is developing generative models that produce three-dimensional geological representations conditioned on geophysical surveys, borehole measurements, and other physical observations. Like image-generation systems that produce different plausible images from a prompt, our models aim to produce plausible views of the subsurface consistent with the evidence available. The goal is not to pretend we know exactly what lies underground. It is to capture both what we know and what remains uncertain, giving exploration teams a stronger foundation for deciding where and how to investigate critical resources. In this role, you’ll help lead the research and development of these models, from exploring architectures and conditioning methods to training, evaluation, and adaptation to real projects. You’ll work alongside AI researchers, engineers, geoscientists, and project teams to connect scientific advances with consequential real-world decisions. Core responsibilities Design, implement, train, evaluate, and iterate on diffusion models that generate 3D geological models. Develop approaches to condition model outputs on geophysical surveys, borehole measurements, and other physical observations. Shape synthetic-data generation and data-curation approaches to improve model quality, robustness, and generalization. Form hypotheses, design rigorous experiments, and evaluate model behavior, uncertainty, and limitations using reproducible training workflows. Adapt generative modeling appro