Software Engineer, AI Engineer, Machine Learning​/ ML Engineer

Join to apply for the Software Engineer role at Terra AI

Get AI-powered advice on this job and more exclusive features.

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.

As a Software Engineer at Terra AI, you will be a key member of a cross‑functional team, collaborating with geologists, reservoir engineers, and ML engineers to configure and deploy large‑scale data processing and ML workloads. This fast‑paced environment offers a significant opportunity to refine your skills and make a large impact on a growing organization.

• Contribute to the development of large‑scale data processing applications.
• Partner with data scientists and ML engineers to operationalize machine learning workloads.
• Design and implement scalable data systems.
• Analyze and optimize performance for distributed workloads, focusing on cost, runtime, and resource efficiency.
• Configure cloud infrastructure to support data‑intensive applications.
• Provide guidance on software development best practices.

• B.S. in computer science or related field
• 5+ years of experience developing data‑intensive software systems using languages like Python or Julia.
• 3+ years of experience designing and building solutions on cloud platforms such as AWS, Azure, or GCP.
• Proven experience packaging, testing, and deploying containerized workloads using tools like Docker and Kubernetes.
• Experience tuning system performance, including optimization for cost, runtime, and resource efficiency.
• Experience working effectively with multidisciplinary teams.
• Demonstrated experience ramping up on a complex industry or highly technical domain.
• Strong drive to learn and engage with new technologies.

• Experience managing orchestration for multi‑step batch workloads.
• Experience with data governance and metadata management tools.
• Experience managing or working with MLOps tooling.
• Experience working with numerical simulation tools.

Mid‑Senior level

Full‑time

Engineering and Information Technology

Services for Renewable Energy

Referrals increase your chances of interviewing at Terra AI by 2x

San Francisco, CA