Terra AI · 22 hours ago
Full Stack Engineer, Scientific Modeling Tools
Redwood City, CA
Full-time
Onsite
Mid LevelTerra AI is building a state-of-the-art AI platform for clean energy and mineral resources. The role involves professionalizing and extending internal modeling tools while collaborating with domain experts to develop usable and extensible software solutions.
Artificial Intelligence (AI)EnergyNatural Resources
Responsibilities
Collaborate closely with domain experts to translate requirements into software that is correct, usable, and extensible
Own and improve internal modeling stacks, including:
Refactoring and modularization for clarity and reuse
Testing strategies that match scientific software realities (golden tests, invariants, property-based testing where useful)
Performance profiling and optimization where it matters
Documentation and developer experience improvements
Design and implement APIs and interfaces that turn working examples into maintainable components
Build configuration management patterns that make runs reproducible and debuggable
Implement and maintain orchestration pipelines for simulation ensembles and data validation
Establish versioning and release practices for internal tooling, including metadata management
Work primarily in Python and Julia
Integrate with ML-adjacent components and artifacts (inputs, outputs, model wrappers), without being responsible for inventing new ML methods
Qualification
Required
Strong software engineering fundamentals and proven ability to take ownership of complex codebases
Production-grade Python skill
Comfort working in Julia or willingness to go deep quickly
Experience designing APIs, handling configuration, and building reliable execution paths for complex workflows
Familiarity with performance profiling and optimization tooling
Familiarity with ML frameworks at an integration level (PyTorch preferred, TensorFlow or JAX also relevant), including artifacts, I/O, and runtime concerns
Experience with orchestration or workflow tooling (Flyte, Prefect, Dagster, or similar), or equivalent patterns built in-house
Preferred
Geophysics or geomodeling experience, including survey simulation or related tooling (SimPEG or similar)
Reservoir simulation experience (Eclipse, Intersect, JutulDarcy, or similar)
Experience solving PDE-based problems in HPC environments
Familiarity with Fortran or C++ codebases common in scientific stacks
Experience in simulation, CAD, CFD, or other engineering/scientific software domains
Experience supporting scientific users and workflows, where communication and shared language matter
Experience with batch pipelines and data-intensive systems
Company
Terra AI
Intelligent Systems for Natural Resource Development
Founded in 2023
2-10 employees
H1B Sponsorship
Terra AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2)
2024 (1)
Funding
Current Stage
Early StageTotal Funding
$3.38MKey Investors
Founders Factory
2025-04-23Convertible Note
2023-10-01Seed· $3.38M
Leadership Team
John Mern
Co-Founder, CEO
Recent News
Company data provided by crunchbase