Terra AI Logo

Terra AI

Staff Machine Learning Engineer

Posted 18 Days Ago
Remote
Hiring Remotely in US
Senior level
Remote
Hiring Remotely in US
Senior level
Lead the development of generative models for 3D geological data, focusing on diffusion models conditioned on geophysical data to improve resource exploration.
The summary above was generated by AI
About Terra AI

We are building the state-of-the-art AI platform for the discovery and development of clean energy and mineral resources. We bring the most advanced techniques in generative AI, foundation modeling, and autonomous decision optimization to tackle the most important problems in the geosciences. These systems can help more reliably identify critical resource deposits, more rapidly measure and characterize them, and design more efficient and sustainable production plans.

We are backed by Khosla Ventures and other leading venture investors. We are now looking to grow our small team to continue to mature our technology and support deployment with our world-class mineral and clean energy partners.

Role description

In the same way image generators have shown the remarkable ability to produce a diverse set of realistic pictures conditioned on a text prompt (and other inputs), we are developing a generative model that produces 3D geological models conditioned on geophysical surveys, bore hole measurements, and other forms of physical observation. The outputs of the generative model capture what we know and don’t know about the state of the subsurface, allowing explorers to make maximally informed decisions about how and where to explore for critical resources. 

We are looking for a talented deep learning engineer or scientist to lead the development of this model that will revolutionize decision making in the earth subsurface for a wide range of clean energy applications.

Role Responsibilities
  • 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. 

Qualifications

Required Qualifications:

  • 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.

Bonus points if you:

  • Experience with Generative Models

    • Familiarity with generative architectures, particularly diffusion models, and an emphasis on posterior sampling methods.

  • Knowledge of Transformer Architectures

    • Experience building and training transformers, especially in applications involving 3D data.

  • Scaling Models Across Large GPU Clusters

    • Expertise in parallelizing models across multiple GPUs and optimizing distributed training pipelines.

  • Cloud Infrastructure Expertise

    • Experience setting up, managing, and optimizing cloud environments for machine learning workloads, including provisioning resources and managing costs.

Top Skills

Cloud Infrastructure
Generative Models
PyTorch
Transformers

Similar Jobs

6 Days Ago
Remote or Hybrid
Santa Clara, CA, USA
164K-286K Annually
Senior level
164K-286K Annually
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
As a Staff AI/ML Software Engineer, you'll build scalable data pipelines, ML models, and integrate AI/ML technologies to enhance customer experiences and automate workflows while collaborating cross-functionally.
Top Skills: AWSAzureBigQueryGCPJavaKafkaKubernetesLightgbmPythonPyTorchScalaSparkSpring BootTensorFlowXgboost
Yesterday
Remote or Hybrid
Santa Clara, CA, USA
173K-303K Annually
Senior level
173K-303K Annually
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
The role involves developing and improving Large Language Models by applying AI/ML techniques, collaborating with teams, and producing innovative solutions for clients.
Top Skills: AIDeep LearningLarge Language Models (Llms)Machine LearningPython
Yesterday
Remote or Hybrid
Santa Clara, CA, USA
164K-286K Annually
Senior level
164K-286K Annually
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
The role involves ensuring the quality of AI and ML products by implementing test strategies, frameworks, and metrics, while collaborating with cross-functional teams.
Top Skills: AIJavaJunitMlSeleniumTestngUnix

What you need to know about the Los Angeles Tech Scene

Los Angeles is a global leader in entertainment, so it’s no surprise that many of the biggest players in streaming, digital media and game development call the city home. But the city boasts plenty of non-entertainment innovation as well, with tech companies spanning verticals like AI, fintech, e-commerce and biotech. With major universities like Caltech, UCLA, USC and the nearby UC Irvine, the city has a steady supply of top-flight tech and engineering talent — not counting the graduates flocking to Los Angeles from across the world to enjoy its beaches, culture and year-round temperate climate.

Key Facts About Los Angeles Tech

  • Number of Tech Workers: 375,800; 5.5% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Snap, Netflix, SpaceX, Disney, Google
  • Key Industries: Artificial intelligence, adtech, media, software, game development
  • Funding Landscape: $11.6 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Strong Ventures, Fifth Wall, Upfront Ventures, Mucker Capital, Kittyhawk Ventures
  • Research Centers and Universities: California Institute of Technology, UCLA, University of Southern California, UC Irvine, Pepperdine, California Institute for Immunology and Immunotherapy, Center for Quantum Science and Engineering

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account