# Senior/Staff Machine Learning Researcher at Terra AI

- Company: Terra AI
- What the company does: A geoscience frontier company building AI to accelerate subsurface resource development for a clean energy future. Backed by Khosla.
- Company website: https://terraai.com/
- Type: Startups (AI role)
- Level: Senior
- Location: US remote
- Work setup: Remote
- Posted: 2026-04-14
- Apply by: 2026-10-08
- Apply: https://jobs.ashbyhq.com/terraai/9f64ab37-62b3-405b-9952-4d63ea35cb57
- Page: https://www.1752.vc/careers/jobs/terra-ai-senior-staff-machine-learning-researcher/

## 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

Tags: Engineering