Terra AI
Senior/Staff Machine Learning Researcher
$175K–$265K
US remote · Remote · Staff · Full-time · Posted Apr 14, 2026
About the job
Leads research and development of generative models that create uncertainty-aware 3D geological representations from geophysical, borehole, and other physical observations.
What you'll do
- Design, train, evaluate, and improve diffusion models for generating 3D geological models.
- Develop conditioning, synthetic-data, and data-curation methods; assess model behavior, uncertainty, and limitations through reproducible experiments.
- Adapt research to real-world projects and develop maintainable, reproducible research code with engineering and geoscience teams.
What they ask for
- 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 for creating, cleaning, validating, and maintaining machine-learning datasets.
- 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 with transformer architectures for 3D or structured data, multi-GPU or cluster training, or cloud ML infrastructure.
Pay
$175,000–$265,000 USD annual base salary
Tools
PyTorch
AI focus
Generative AI research for 3D geological modeling, including diffusion models conditioned on physical observations and methods for representing uncertainty.
Model-written summary · the original posting is authoritative