> Markdown version of [/jobs/ext/4694903-senior-staff-machine-learning-researcher](https://www.wearedevelopers.com/jobs/ext/4694903-senior-staff-machine-learning-researcher). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Senior/Staff Machine Learning Researcher

- **Company:** Terra AI, Inc.
- **Location:** Redwood City, CA, United States
- **Experience:** Expert
- **Salary:** $175,000 - $265,000
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, Cloud Computing, Computer Clusters, Software Code Optimization, Software Debugging, Machine Learning, Software Engineering, Pytorch, Deep Learning, Software Version Control, Data Generation
- **Published:** November 12, 2026
- **Apply:** https://www.careerbuilder.com/job-details/senior-staff-machine-learning-researcher-redwood-city-ca--a5e1c72b-52ce-420a-b135-2293b170849d

## About the Role

Machine Learning, Research Skills, Artificial Intelligence (AI), Earth Sciences, Category Development, Problem Solving Skills, Mining Industry, Underground Mining, Geology, Drilling, 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., * Extensive hands-on PyTorch experience, including custom modules, training optimization, and debugging complex or large-scale models.

 * Demonstrated experience designing, implementing, and training substantial deep learning architectures from scratch.
 * Strong data-curation skills, including creating, cleaning, validating, and maintaining datasets for machine learning.
 * Solid software engineering and system-design fundamentals, including version control, testing, code optimization, scalability, and maintainability.

Nice to have

 * Experience with generative architectures, especially diffusion models and posterior sampling methods.
 * Experience developing and training transformer architectures, particularly for 3D or other structured data.
 * Experience scaling model training across multi-GPU systems or large GPU clusters, including distributed training optimization.
 * Experience managing and optimizing cloud infrastructure for machine learning workloads.

We recognize that exceptional candidates come from a range of research, engineering, and scientific backgrounds. If this work excites you and your experience aligns with the spirit of the role, we encourage you to apply even if you do not meet every preferred qualification.

## Description

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 approaches to real-world projects in close collaboration with geoscience and delivery teams.
 * Develop clear, maintainable research code and collaborate with engineering teams to make promising approaches reproducible and useful beyond a single experiment.