AI/ML Engineer [Senior-Principal]
Summary
Location: Sunnyvale, CA (Hybrid)
Employment Type: Full-Time
Job Category: AI/ML Engineering
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.
About the Role
We are hiring a [Senior - Principal] AI/ML engineer to develop several elements of our AI/ML platform, including 3D generative modeling, neural surrogate modeling, geospatial predictive modeling, and deep reinforcement learning. This role will work with AI/ML research scientists to identify model architectures, prepare data pipelines, and train models on multi-GPU cloud infrastructure.
Key responsibilities include:
Working with our founding team to define the technical roadmap for prioritizing and developing key deep learning components
Design and training of neural network models for 3D physics simulation, 3D geology generation, resource prediction, and development optimization.
Development of foundation models with mixtures of synthetic and real-world data.
Refining pipelines for data ingestion, validation and versioning.
Continually improving model validation, deployment and versioning.
Collaborate with the software engineering team members to integrate deep learning technology into the product backend.
Play a critical role in hiring and building up our technical team.
We are looking for someone who
Is excited about our mission to improve the speed and adoption of critical technologies to drive to net zero by 2050.
Can drive forward in ambiguous environments.
Has start up experience in building a full end to end web application product across back end, cloud services and front end.
Can balance trade-offs and decisions around utilizing micro services vs building ourselves.
Is curious to learn about our customers, their workflows and applications of our product and the underlying AI and ML technology.
Is an excellent communicator and team player
Typical Qualifications (Not Requirements)
BS in computer science or related field with 5 years engineering/research experience
MS/PhD in computer science or related field with 2 years engineering/research experience
Experience working with multidisciplinary teams (across UI/UX design, product, AI, ML)
Demonstrated proof of building deep learning models from scratch
Ability to come to our co-working location in Sunnyvale, CA 3 days per week
Why should you join us?
This is an incredible opportunity to be part of the foundation of a company and product. You will get to work with a fun team to deliver an incredibly valuable end to end product. You will receive a significant equity stake in the company. Additionally we offer a suite of benefits and perks, including
Company-subsidized medical, dental, & vision insurance plans.
Mental health support (if not covered by medical insurance plans)
401k plan with company matching
Flexible PTO plus company holidays
Annual company-wide winter break (December 24 - January 1)
Professional development budget to attend conferences or other events
One-time home-office set up stipend
Paid parental leave
Terra AI is an equal opportunity employer. At Terra AI, we are committed to building a diverse and inclusive workplace, because inclusion and diversity are essential to achieving our mission. We encourage you to apply if you are excited about Terra and this role even if your past experience does not align perfectly with every qualification in the job description.