Senior Staff Machine Learning Researcher

Terra

Senior–Staff Machine Learning Researcher at Terra AI developing generative models for clean energy and mineral resource exploration. Leading deep learning model development and collaboration with engineering teams.

Posted 4/14/2026full-timeRemote • 🇺🇸 United StatesSeniorWebsite

Tech Stack

Tools & technologies

PyTorch

About the role

Key responsibilities & impact

Design, train, test, and iterate on diffusion models for 3D geological models
Design, train, test, and iterate on an approach to for conditioning generation on geophysical data and other observations
Inform the generation of synthetic data to improve model performance
Adapt diffusion modeling approach to specific real-world projects in collaboration with project teams.

Requirements

What you’ll need

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.
Strong Software Engineering and Design Experience
Proficient in software development best practices, including version control, testing, and code optimization.
Familiarity with designing scalable and maintainable systems.

Benefits

Comp & perks

Health insurance
Flexible work arrangements

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

PyTorchdeep learningdiffusion modelsdata curationsoftware engineeringcustom modulestraining optimizationdebuggingcomplex architecturesversion control

Soft Skills

collaborationproblem-solvingadaptabilitycommunicationcreativity